跳到正文
原文
thexpin· @thexpin · X·· 5 天前AI 评分72

DeepSeek 发布与 Nvidia 组件对应的 Ascend 工具套件

AI 导读

DeepSeek 发布了与 Nvidia 组件对应的 Ascend 工具套件,包括 TileLang、DeepGEMM、DeepEP、TileKernels、FlashMLA 和 DeepSelect。

正文 · AI 翻译

DeepSeek 发布了与 Nvidia 组件对应的昇腾工具包:TileLang、DeepGEMM、DeepEP、TileKernels、FlashMLA 和 DeepSelect。

它表示,其训练中使用的每个 TileLang 内核现在都有了高性能的昇腾实现。V4 系列训练中的大多数内核都由 TileLang 驱动。

DeepSeek 和华为还联合优化了一个 128 卡的昇腾 950 超节点,据称关键计算和通信测试已接近硬件极限。

- TileLang: github.com/tile-ai/tilelang

- DeepGEMM Ascend: github.com/deepseek-ai/DeepG…

- DeepEP Ascend: github.com/deepseek-ai/DeepE…

- TileKernels: github.com/deepseek-ai/TileK…

- FlashMLA: github.com/deepseek-ai/Flash…

- DeepSelect: github.com/deepseek-ai/DeepS…

来源:thexpin · x.com