tilelang

ChainCatcher 消息，人工智能公司 DeepSeek 正式开源面向华为昇腾算力平台的基础设施组件，涵盖 TileLang 高级语言编译工具、计算库和分布式通信库，并与此前面向英伟达平台的开源组件一一对应。TileLang 旨在提供通用、编程更简单且能达到硬件性能上限的高级语言，相对 CUDA 可提高开发效率并简化逻辑，其编程模型也可发挥芯片特性。TileLang 路线先在英伟达平台验证，并已承载 DeepSeek V4 系列模型训练中大部分算子的实现。本次开源的昇腾版本对昇腾 Ascend C 底层指令进行封装，在不损失硬件性能的情况下提供高级语言编程方式。目前，DeepSeek 训练中用到的每一个 TileLang 算子，在昇腾上都有对应的高性能实现。同期开源的组件还包括 DeepGEMM、DeepEP、TileKernels、FlashMLA 和 DeepSelect，分别用于通用矩阵运算、大规模跨设备通信、常规向量计算与访存、长上下文稀疏注意力以及数据筛选。DeepSeek 称，多项关键测试中相关计算与通信性能已接近硬件上限；研发过程中华为团队给予支持，双方合作推进基于昇腾 950 的 128 卡超节点方案，并对计算与通信做了深度优化。