DeepSeek 于9月30日开源了面向华为昇腾950平台的全套底层基础设施组件,涵盖TileLang编译工具及DeepGEMM、DeepEP、TileKernels、FlashMLA、DeepSelect五类计算与通信库,与此前英伟达平台开源组件一一对应。其中TileLang昇腾后端支持原生代码生成、自动调度与同步,已承载DeepSeek V4训练中大部分算子;各组件上层接口与英伟达版本保持兼容,可降低迁移成本。DeepSeek还与华为联合完成128卡超节点方案的计算与通信协同优化。此次开源为国产AI芯片平台提供了从算子编写到计算、通信执行的完整开发入口,但部分组件在数据格式、算子覆盖范围上仍与CUDA版本存在差异,实际部署效果取决于与上层框架及集群环境的集成情况。
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅