摩尔线程成功支持 DeepSeek 开源通信库 DeepEP 和并行算法 DualPipe
2025-03-03 15:10:02 小编:联水游戏园
本站 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》
- 猜你喜欢
-
九龙趣看免费下载9.9MB手游下载
-
浮空秘境手机版1291.57M手游下载
-
雪豹视频9.9MB手游下载
-
梦回凤歌手游330.41M手游下载
-
红叶影评9.9MB手游下载
-
愤怒的小鸟变形金刚安卓版493.56M手游下载
-
精灵大夜战手游565.81M手游下载
-
特狗app下载官网版9.9MB手游下载
-
小猫钓鱼安卓14.48M手游下载
- 相关手机游戏
-
好好看免费追剧APP9.9MB手游下载
-
幻龙战记安卓版633.00M手游下载
-
豪华曹魏传app安卓版804.90M手游下载
-
山海视频app免费追剧9.9MB手游下载
-
星空影视app最新版9.9MB手游下载
-
达达兔影视2025最新版9.9MB手游下载
-
山茶视频app免费追剧不看广告9.9MB手游下载
-
影视仓app最新版9.9MB手游下载
-
影视森林3.0.0版本9.9MB手游下载
- 推荐手游资讯
- 最新手机精选
-
植物大战僵尸杂交版内置菜单2025-01-09手游下载 | 403.23MB
-
地铁跑酷火山难度版2025-02-21手游下载 | 57.33MB
-
弓箭传说2无限内购版2024-11-30手游下载 | 389.9MB
-
印度摩托车驾驶3D2025-02-05手游下载 | 135.94M
-
三国冰河时代0.1折2025-02-18手游下载 | 245.65MB
-
梅特尔恐怖逃生中文版2025-02-18手游下载 | 256.00MB