DeepSeek开源昇腾版训练组件,国产算力承载主力训练
DeepSeek训练栈整体移植昇腾并公开代码,国产算力能否承载前沿训练首次有了栈级验证。
DeepSeek于9月30日通过官方公众号宣布,开源面向华为昇腾平台的训练基础设施组件,与此前英伟达版本一一对应。
本次发布包括TileLang编译工具、DeepGEMM矩阵运算库、DeepEP跨设备通信库,以及TileKernels、FlashMLA和DeepSelect,代码已在GitHub公开。公司称V4系列模型训练中大部分算子已由TileLang实现,且每个算子在昇腾上都有对应版本。
“接近硬件上限”“不损失性能”为DeepSeek自述,摘要未附第三方测试。但训练栈整体移植并公开代码本身,说明昇腾已能承载其主力训练负载,这是国产算力生态迄今最直接的栈级验证。
信息源:https://github.com/deepseek-ai/DeepGEMM-Ascendhttps://github.com/deepseek-ai/DeepEP-Ascendhttps://www.ithome.com/1/008/604.htm