DeepSeek 开源昇腾基础组件:TileLang 等与英伟达版一一对应

9 月 30 日,DeepSeek 通过官方公众号宣布,正式开源面向华为昇腾算力平台的基础设施组件,涵盖 TileLang 高级语言编译工具、计算库和分布式通信库。据官方介绍,所有组件与此前面向英伟达平台的开源组件一一对应。

要建立新一代自主可控的 GPU 软件生态,首要任务是建立一个通用、编程简单、同时又能达到硬件性能上限的高级语言。据公开信息,TileLang 正是在这一背景下诞生,由北京大学团队发起,采用 Python 式语法,让开发者能够专注于提高生产力,而无需牺牲实现最佳性能所需的底层优化。

TileLang:让昇腾也能用上高级语言

据官方介绍,相对于英伟达的 CUDA 语言,TileLang 编程更简单,能显著提高开发效率、简化代码逻辑;相对于其他同类高级语言,其编程模型可以充分发挥芯片特性,达到硬件性能上限。官方称,TileLang 路线首先在英伟达成熟平台上得到验证,目前已承载 DeepSeek V4 系列模型训练中大部分算子的实现,是开发高性能算子的核心工具。

本次开源的 TileLang 昇腾版本,对昇腾 Ascend C 底层指令进行封装,提供高级语言的编程方式,同时不损失硬件性能。据官方介绍,目前 DeepSeek 训练中用到的每一个 TileLang 算子,在昇腾上都有对应的高性能实现。

图:华为昇腾是本次组件开源的适配目标平台
打开网易新闻 查看精彩图片
图:华为昇腾是本次组件开源的适配目标平台

计算与通信组件:五类核心库

本次开源还包括昇腾平台的核心计算与通信组件,为不同使用场景提供可复用的基础能力:

- DeepGEMM:加速通用矩阵运算

- DeepEP:提供高效的大规模跨设备通信

- TileKernels:提供数据处理所需的常规向量计算和访存算子

- FlashMLA:提供稀疏注意力算子,提升长上下文处理效率

- DeepSelect:实现高效的数据筛选

据官方介绍,在多项关键测试用例中,上述组件的计算与通信性能已接近硬件上限。官方同时汇总了开源项目地址,包括 TileLang、DeepGEMM-Ascend、DeepEP-Ascend 等仓库。

图:面向昇腾的组件与英伟达版一一对应
打开网易新闻 查看精彩图片
图:面向昇腾的组件与英伟达版一一对应

为什么值得关注

对开发者而言,这批组件的价值在于「可复用的基础能力」。过去在非英伟达平台上复现大模型训练与推理,往往需要围绕底层指令重新做大量适配;把编译工具、计算库与通信库成套开源,等于把适配昇腾的一整套工程链路交到开发者手里,降低了迁移与调优的重复劳动。

官方还强调,TileLang 昇腾版本是一个开源工作,希望能对更多 AI 芯片建立高可用软件生态起到示范作用。这意味着其目标并不局限于单一平台,而是试图以统一的高级语言抽象,降低不同芯片之间软件栈的适配成本。

图:高级语言抽象被视为降低芯片适配成本的关键
打开网易新闻 查看精彩图片
图:高级语言抽象被视为降低芯片适配成本的关键

展望

随着国产算力在训练与推理中的使用比例提高,软件栈的完善程度将直接影响模型的落地效率。DeepSeek 这次把面向昇腾的基础设施组件成套开源,能否带动更多开发者围绕统一的高级语言做出可复现的成果,是接下来值得观察的方向。

关键词: DeepSeek、昇腾、TileLang、开源、国产算力