字号 ·· | 护眼
凤凰网科技

DeepSeek开源升腾基础组件:与英伟达平台一一对应

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

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

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

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