DeepSeek开源昇腾全栈工具链,AI硬件生态竞争升级

9月30日,AI研发公司DeepSeek做出了一项可能影响行业格局的举动:正式将其为华为昇腾AI算力平台开发的全套基础设施组件开源。这一动作不仅为昇腾生态开发者带来了与英伟达平台对等的软件工具选择,也标志着AI基础软件层的竞争进入了新的阶段。

从编译到通信:一套完整的昇腾开发解决方案

本次开源的核心是三个相互协同的组成部分:TileLang高级语言编译工具、一系列核心计算库以及专为大规模训练优化的分布式通信库。它们共同构成了在昇腾平台上进行高效AI应用开发的完整技术栈。

TileLang:让昇腾芯片编程更“高级”

针对昇腾平台,TileLang编译工具扮演了关键角色。它通过对昇腾Ascend C底层指令进行高级封装,允许开发者使用更抽象、更易用的编程方式编写代码,同时通过编译优化确保生成的程序能充分发挥硬件性能,避免了高级语言常带来的性能损耗。

五大核心组件,覆盖AI计算关键场景

除了编译工具,开源组件库还包含了五个针对不同计算模式优化的核心模块:

  • DeepGEMM:专门优化通用矩阵乘法运算,这是深度学习模型中最基础也最耗时的操作之一。
  • DeepEP:设计用于大规模分布式训练场景,提供高效的跨设备、跨节点通信能力,减少训练中的等待时间。
  • TileKernels:提供数据处理所需的各种常规向量计算和数据存取算子,是构建更复杂操作的基础。
  • FlashMLA:针对当前大模型的热点——稀疏注意力机制进行优化,能显著提升长文本、长序列等上下文处理任务的效率。
  • DeepSelect:实现了高效的数据筛选操作,对于数据预处理和特定模型推理至关重要。

性能表现:逼近硬件理论极限

根据DeepSeek公布的测试结果,这套开源组件在多项关键基准测试中表现突出,其计算效率和通信带宽利用率已经非常接近昇腾硬件本身的理论性能上限。这意味着开发者无需进行繁琐的底层优化,就能获得近乎极致的硬件性能,大幅降低了在异构算力平台上开发高性能AI应用的门槛。

此次开源可以看作是DeepSeek对其技术战略的一次清晰展示。通过为不同的主流AI芯片平台提供性能对标、功能完备的开源软件栈,它正试图在AI基础软件领域建立起新的行业标准。对于广大AI开发者和企业而言,这无疑增加了其在芯片选型上的灵活性和主动权。