跳到正文
硅脉动SiliconMai

DeepSeek开源昇腾组件,覆盖编译与通信工具

新组件与此前英伟达平台版本对应,TileLang封装Ascend C底层指令。DeepSeek称已为训练使用的每个TileLang算子提供昇腾实现。

硅脉动
DeepSeek资料图

DeepSeek9月30日开源面向华为昇腾平台的基础设施组件,包括TileLang高级语言编译工具、计算库和分布式通信库。公司表示,这些组件与此前面向英伟达平台开源的工具一一对应,扩大了其训练工具对不同算力平台的支持。

此次昇腾版TileLang对Ascend C底层指令进行封装,为开发者提供高级语言编程方式。DeepSeek称,这种设计可以简化代码逻辑,同时保留对芯片性能的利用能力。有关效率和性能的描述来自公司介绍。

TileLang此前先在英伟达平台验证,目前已承载DeepSeek V4系列模型训练中大部分算子的实现。算子是模型运行中的具体计算操作,此次平台适配涉及这些操作在另一套硬件上的实现。

DeepSeek表示,训练使用的每一个TileLang算子,目前在昇腾上都有对应的高性能实现。开源范围因此同时包括编译工具、计算及通信环节,而非只提供单项语言工具。

公司将TileLang定位为开发高性能算子、探索模型训练方式的核心工具,强调高级语言编程模型与硬件特性之间的结合。其介绍中对CUDA和其他高级语言的效率比较,属于DeepSeek自身的技术评价。

研发过程中,DeepSeek与华为团队共同推进基于昇腾950的128卡超节点方案,并对计算和通信进行优化。华为参与了面向昇腾平台的适配工作,相关合作也涵盖超节点层面的计算与通信设计。