thexpin· @thexpin · X·· 5 天前AI 评分
DeepSeek 发布与 Nvidia 组件对应的 Ascend 工具套件
DeepSeek 发布了与 Nvidia 组件对应的 Ascend 工具套件,包括 TileLang、DeepGEMM、DeepEP、TileKernels、FlashMLA 和 DeepSelect。
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