DeepSeek 重磅开源:面向华为昇腾的 AI 基础设施组件,打破 CUDA 生态垄断
当全球 AI 开发者还在为 NVIDIA CUDA 生态依赖而苦恼时,DeepSeek 迈出了关…
文章目录
当全球 AI 开发者还在为 NVIDIA CUDA 生态依赖而苦恼时,DeepSeek 迈出了关键一步。今天,DeepSeek 正式开源面向华为昇腾算力平台的基础设施组件,涵盖 TileLang 高级语言编译工具、计算库、分布式通信库。所有组件与此前面向英伟达平台的开源组件一一对应。这不仅是技术突破,更是中国 AI 产业自主可控的重要里程碑。
一分钟速览
- DeepSeek 开源面向华为昇腾算力平台的基础设施组件
- 涵盖 TileLang 高级语言编译工具、计算库、分布式通信库
- 所有组件与英伟达平台开源组件一一对应
- TileLang 编程更简单,同时达到硬件性能上限
- 已承载 DeepSeek V4 系列模型训练中大部分算子的实现
为什么这次开源如此重要?
在全球 AI 算力竞争中,NVIDIA 凭借 CUDA 生态占据了主导地位。CUDA 不仅是编程语言,更是一个完整的软件生态,包括编译器、调试器、性能分析工具等。这种生态优势,让 NVIDIA 在 AI 芯片市场形成了近乎垄断的地位。
DeepSeek 此次开源的昇腾基础组件,直接挑战了这一格局。通过提供与 CUDA 对应的开源工具链,DeepSeek 让开发者可以在华为昇腾芯片上实现与 NVIDIA 平台相当的开发体验。这意味着:
降低迁移成本:开发者可以更容易地将现有 CUDA 代码迁移到昇腾平台。
打破生态垄断:为 AI 芯片市场引入更多竞争,推动技术进步。
自主可控:中国 AI 产业可以减少对国外芯片和软件的依赖。
核心组件详解
本次开源的组件包括:
1. TileLang 高级语言编译工具
TileLang 是 DeepSeek 自研的高级语言,相对于 NVIDIA 的 CUDA 语言,编程更简单,能显著提高开发效率、简化代码逻辑。同时,相对于其它同类高级语言,TileLang 的编程模型可以充分发挥芯片的特性,达到硬件的性能上限。
TileLang 路线首先在英伟达成熟平台上得到了验证,如今已承载了 DeepSeek V4 系列模型训练中大部分算子的实现,是 DeepSeek 探索 AGI 新范式、开发高性能算子的核心工具。
本次开源的 TileLang 昇腾版本,对昇腾 Ascend C 底层指令进行封装、提供高级语言的编程方式、同时不损失硬件性能。这正是 TileLang 的设计初衷。
2. DeepGEMM 加速通用矩阵运算
提供高效的通用矩阵乘法(GEMM)运算,这是深度学习中最核心的计算操作之一。
3. DeepEP 高效大规模跨设备通信
提供高效的大规模跨设备通信能力,支持分布式训练场景。
4. TileKernels 常规向量计算和访存算子
提供数据处理所需的常规向量计算和访存算子,是基础计算组件。
5. FlashMLA 稀疏注意力算子
提供稀疏注意力算子,提升长上下文处理效率。这对于大语言模型处理长文本至关重要。
6. DeepSelect 高效数据筛选
实现高效的数据筛选,优化数据加载和预处理流程。
性能表现:接近硬件上限
据 DeepSeek 介绍,在多项关键测试用例中,上述组件的计算与通信性能已接近硬件上限。这意味着,使用这些开源组件,开发者可以充分发挥昇腾芯片的性能潜力。
特别值得一提的是,目前 DeepSeek 训练中用到的每一个 TileLang 算子,在昇腾上都有对应的高性能实现。这证明了这些组件在实际生产环境中的可靠性。
华为的大力支持
在面向昇腾平台的研发过程中,华为团队给予了毫无保留的大力支持。双方紧密合作,共同推进基于昇腾 950 的 128 卡超节点方案、共同对计算与通信进行了深度优化。
这种紧密合作,反映出中国 AI 产业上下游协同创新的良好态势。DeepSeek 提供算法和软件能力,华为提供芯片和硬件支持,共同推动中国 AI 产业的发展。
开源链接
本次开源的项目链接:
- TileLang: https://github.com/tile-ai/tilelang
- DeepGEMM Ascend: https://github.com/deepseek-ai/DeepGEMM-Ascend
- DeepEP Ascend: https://github.com/deepseek-ai/DeepEP-Ascend
- TileKernels: https://github.com/deepseek-ai/TileKernels
- FlashMLA: https://github.com/deepseek-ai/FlashMLA
- DeepSelect: https://github.com/deepseek-ai/DeepSelect
行业影响:AI 芯片生态的新变量
DeepSeek 此次开源,对 AI 芯片生态具有深远影响:
为国产 AI 芯片提供软件生态:软件生态一直是国产 AI 芯片的短板。DeepSeek 的开源组件,为昇腾芯片提供了高质量的软件工具链,有助于提升其市场竞争力。
推动 AI 算力多元化:随着更多开源组件的出现,开发者可以更容易地在不同芯片平台之间切换,避免被单一供应商锁定。
促进技术创新:开源社区可以基于这些组件进行改进和创新,推动整个 AI 软件生态的发展。
结语
DeepSeek 开源昇腾基础组件,是中国 AI 产业发展的重要一步。通过提供高质量的开源工具链,DeepSeek 不仅降低了开发者的迁移成本,更为国产 AI 芯片的发展提供了有力支持。在全球 AI 竞争日益激烈的背景下,这种自主可控的技术积累,将成为中国 AI 产业的重要竞争力。未来,我们期待看到更多类似的开源项目,共同推动 AI 技术的进步和普及。
来源:DeepSeek 官方微信公众号
本文地址:https://www.163264.com/15794