DeepSeek开源昇腾基础组件:与英伟达平台一一对应
1 小时前 / 阅读约2分钟
来源:凤凰网
DeepSeek宣布开源面向华为昇腾算力平台的基础设施组件,包括TileLang等,性能接近硬件上限,华为团队给予支持,相关项目已在GitHub开源。

凤凰网科技讯 9月30日,DeepSeek宣布正式开源面向华为昇腾算力平台的基础设施组件,涵盖TileLang高级语言编译工具、计算库及分布式通信库。所有组件与此前面向英伟达平台的开源组件一一对应。

其中,TileLang昇腾版本对昇腾Ascend C底层指令进行封装,提供高级语言编程方式,同时不损失硬件性能。DeepSeek称,其训练中用到的每一个TileLang算子,在昇腾上都有对应的高性能实现。TileLang路线此前已在英伟达平台得到验证,承载了DeepSeek V4系列模型训练中大部分算子的实现。

此次同步开源的核心计算与通信组件还包括:DeepGEMM(加速通用矩阵运算)、DeepEP(大规模跨设备通信)、TileKernels(向量计算与访存算子)、FlashMLA(稀疏注意力算子)及DeepSelect(数据筛选)。多项关键测试中,上述组件的计算与通信性能已接近硬件上限。

DeepSeek表示,研发过程中华为团队给予大力支持,双方共同推进基于昇腾950的128卡超节点方案,并对计算与通信进行了深度优化。相关项目已在GitHub开源。