DeepSeek 通过官方公众号宣布,正式开源面向华为昇腾算力平台的基础设施组件,涵盖 TileLang 高级语言编译工具、计算库及分布式通信库,所有组件与此前面向英伟达平台的开源组件一一对应。
TileLang 相对英伟达 CUDA 语言编程更简单,可显著提高开发效率并简化代码逻辑,其编程模型能充分发挥芯片特性以达到硬件性能上限。该路线已在英伟达成熟平台验证,目前承载 DeepSeek V4 系列模型训练中大部分算子的实现。
本次开源的 TileLang 昇腾版本对 Ascend C 底层指令进行封装,提供高级语言编程方式且不损失硬件性能。开源组件还包括 DeepGEMM、DeepEP、TileKernels、FlashMLA 和 DeepSelect,在多项关键测试用例中计算与通信性能已接近硬件上限。
文中提及的 AI 工具
评论
正在加载评论…