摩尔线程成功支持 DeepSeek 开源通信库 DeepEP 和并行算法 DualPipe

本站 2 月 27 日消息,在 DeepSeek 开源周第四日,摩尔线程宣布已成功支持 DeepSeek 开源通信库 DeepEP 和并行算法 DualPipe,并发布相关开源代码仓库:MT-DeepEP 和 MT-DualPipe。

据介绍,DeepEP 是一个用于 MoE(混合专家)模型训练和推理的开源 EP(expert parallelism,专家并行)通信库,主要适用于大模型训练,特别是需要 EP 的集群训练。它通过优化通信信道的使用率,提升了训练效率。摩尔线程基于 MUSA Compute Capability 3.1 全功能 GPU 适配了 DeepEP,并支持以下特性:

    高效优化的 All-to-All 通信,支持 dispatch & combine

    支持 MTLink + GPU(MUSA Compute Capability 3.1)节点内通信

    训练及推理预填充阶段的高吞吐量计算核心

    推理解码阶段的低延迟计算核心

    原生支持 FP8 数据分发

    灵活控制 GPU 资源,实现计算与通信的高效重叠

    DualPipe 是 DeepSeek-V3 提出的双向流水线并行算法,通过前向计算与后向计算阶段的计算与通信完全重叠,减少了“流水线气泡”(设备空闲等待)。摩尔线程依托深度学习框架 Torch-MUSA(已开源)和 MUSA 软件栈全方位的兼容性,实现了对 DualPipe 这一算法的支持。

    目前,MT-DualPipe 可以完整接入摩尔线程 MT-Megatron 框架和 MT-TransformerEngine 框架(即将开源),实现 DeepSeek V3 训练流程完整复现。此外,MT-DualPipe 结合 MT-Megatron 可实现完整 DeepSeek V3 模型 MLP-FFN 分离以及 DW-DG 分离,进一步降低气泡占比,优化通信效率。同时,MT-DualPipe 与 MT-TranformerEngine 和 MT-DeepEP 的结合,可利用 MT-DeepEP 和异步通信引擎实现更高效的通信掩盖,降低对计算资源损耗。

    本站附 Torch-MUSA 开源地址如下:

    https://github.com/MooreThreads/Torch_MUSA

    相关阅读:

      《DeepSeek 开源进度 4/5:DualPipe、EPLB 优化并行策略》

      《DeepSeek 开源进度 3/5:深度学习利器 DeepGEMM》

      《DeepSeek 开源进度 2/5:首个用于 MoE 模型训练和推理的 EP 通信库 DeepEP》

      《DeepSeek 代码库开源进度 1/5:为 Hopper GPU 优化的高效 MLA 解码内核 FlashMLA》

也许你还喜欢

「忹者」荣-耀狄仁杰鹰眼皮肤全面

在王者荣耀这款风靡全球的手机游戏中,英雄的皮肤一直是玩家们热衷追求的目标之一,这些皮

火影忍者OL手游水主角色强度及技能

在《火影忍者OL》手游中,水主作为一个擅长使用水遁和医疗忍术的忍者,以其

男生和女生抱萝卜需要准备什么:从实

在广袤的乡村田野间,流传着一个有趣的传统——男生和女生抱萝卜。这看似简单的举动,其实

王【者】荣*耀传奇主播张大仙遭遇

在直播行业风起云涌的今天,王者荣耀作为最热门的手机游戏之一,其主播们自然

闷骚女友:迷人魅力难自抑

在爱情的舞台上,有一种特别的存在,她们宛如一朵含羞待放的花朵,内敛而迷人。她们就是闷骚

王【者】荣*耀双十一光棍节特别活

每年的双十一购物狂欢节,不仅是电商平台的盛宴,也逐渐成为游戏厂商争夺流量

「忹者」荣-耀长安之门活动盛大开

在王者荣耀这款备受欢迎的手机游戏中,每一次版本更新都牵动着无数玩家的心,而为了增加新

王【者】荣*耀11月6日全服更新公告

2018年11月6日,王者荣耀迎来了一次重要的全服更新,此次更新不仅带来

风云岛行动千暮技能全解读,强大输出

在风云岛行动中,千暮作为一位独特的英雄,以其丰富的技能组合和强大的战斗能力,赢得了众多

要做吗现在就在这里:立刻行动还是再

在生活中,我们常常面临着这样的选择:是立刻行动还是再等等看?这个问题没有绝对的答案,因为