DeepSeek 官微宣布,正式开源面向华为昇腾算力平台的基础设施组件,内容涵盖 TileLang 高级语言编译工具、计算库及分布式通信库。这些组件与此前针对英伟达平台的开源版本一一对应。
此次开源的 TileLang 昇腾版,对昇腾的 Ascend C 底层指令进行了封装,在提供高级语言编程方式的同时,严格保障了硬件性能。这正是 TileLang 的设计初衷。目前,DeepSeek 训练过程中使用的每一个 TileLang 算子,在昇腾平台上均已实现了高性能版本。
作为一项开源工程,DeepSeek 希望此举能为更多 AI 芯片构建高可用软件生态树立榜样。本次开源还包括了昇腾平台的核心计算与通信组件,旨在为不同场景提供可复用的基础能力:
DeepGEMM 加速通用矩阵运算;
DeepEP 提供高效的大规模跨设备通信;
TileKernels 提供数据处理所需的常规向量计算和访存算子;
FlashMLA 提供稀疏注意力算子,提升长上下文处理效率;
DeepSelect 则实现了高效的数据筛选。
在多项关键测试用例中,上述组件的计算与通信性能已逼近硬件上限。
