DeepSeek開源周第四日,摩爾線程宣布已成功支持DeepSeek開源通信庫DeepEP和并行算法DualPipe,并發(fā)布相關(guān)開源代碼倉庫:MT-DeepEP和MT-DualPipe。
DeepEP是一個用于MoE(混合專家)模型訓(xùn)練和推理的開源EP(expert parallelism,專家并行)通信庫,主要適用于大模型訓(xùn)練,特別是需要EP的集群訓(xùn)練。它通過優(yōu)化通信信道的使用率,顯著提升了訓(xùn)練效率。摩爾線程基于MUSA Compute Capability 3.1全功能GPU,第一時間適配了DeepEP,支持以下特性:
高效優(yōu)化的 All-to-All 通信,支持 dispatch & combine
支持 MTLink + GPU(MUSA Compute Capability 3.1)節(jié)點(diǎn)內(nèi)通信
訓(xùn)練及推理預(yù)填充階段的高吞吐量計(jì)算核心
推理解碼階段的低延遲計(jì)算核心
原生支持 FP8 數(shù)據(jù)分發(fā)
靈活控制 GPU 資源,實(shí)現(xiàn)計(jì)算與通信的高效重疊
DualPipe是DeepSeek-V3提出的雙向流水線并行算法,通過前向計(jì)算與后向計(jì)算階段的計(jì)算與通信完全重疊,顯著減少了“流水線氣泡”(設(shè)備空閑等待)。與傳統(tǒng)流水線并行相比,DualPipe 采用雙向數(shù)據(jù)流設(shè)計(jì),使數(shù)據(jù)從兩端相向處理,大幅提升了資源利用率與訓(xùn)練效率。
摩爾線程依托深度學(xué)習(xí)框架Torch-MUSA(已開源)和MUSA軟件棧全方位的兼容性,實(shí)現(xiàn)了對DualPipe這一算法的高效支持。MT-DualPipe可以完整接入摩爾線程 MT-Megatron框架和MT-TransformerEngine框架(即將開源),實(shí)現(xiàn)DeepSeek V3訓(xùn)練流程的完整復(fù)現(xiàn)。
此外,MT-DualPipe結(jié)合MT-Megatron可以實(shí)現(xiàn)完整DeepSeek V3模型的MLP-FFN分離以及DW-DG分離,進(jìn)一步降低氣泡占比,優(yōu)化通信效率。同時,MT-DualPipe與MT-TranformerEngine和MT-DeepEP的結(jié)合,可以利用MT-DeepEP和異步通信引擎實(shí)現(xiàn)更高效的通信掩蓋,大大降低對計(jì)算資源的損耗。
DeepEP和DualPipe的第一時間適配,充分驗(yàn)證了摩爾線程MUSA架構(gòu)和全功能GPU在生態(tài)兼容與快速適配上的強(qiáng)大優(yōu)勢。我們誠摯歡迎廣大開發(fā)者下載體驗(yàn),并期待您的反饋與建議。
關(guān)于摩爾線程
摩爾線程成立于2020年10月,以全功能GPU為核心,致力于向全球提供加速計(jì)算的基礎(chǔ)設(shè)施和一站式解決方案,為各行各業(yè)的數(shù)智化轉(zhuǎn)型提供強(qiáng)大的AI計(jì)算支持。
我們的目標(biāo)是成為具備國際競爭力的GPU領(lǐng)軍企業(yè),為融合人工智能和數(shù)字孿生的數(shù)智世界打造先進(jìn)的加速計(jì)算平臺。我們的愿景是為美好世界加速。
-
開源
+關(guān)注
關(guān)注
3文章
4207瀏覽量
46144 -
摩爾線程
+關(guān)注
關(guān)注
2文章
279瀏覽量
6454 -
DeepSeek
+關(guān)注
關(guān)注
2文章
835瀏覽量
3269
原文標(biāo)題:DeepSeek開源|摩爾線程完成對DeepEP和DualPipe的支持
文章出處:【微信號:moorethreads,微信公眾號:摩爾線程】歡迎添加關(guān)注!文章轉(zhuǎn)載請注明出處。
發(fā)布評論請先 登錄
摩爾線程正式開源TileLang-MUSA項(xiàng)目
摩爾線程在SIGGRAPH Asia 2025斬獲3DGS重建挑戰(zhàn)賽銀獎
摩爾線程高開468% 中一簽賺27萬 國產(chǎn)GPU第一股摩爾線程高開468%
摩爾線程發(fā)布Torch-MUSA v2.7.0版本
摩爾線程亮相GOTC 2025全球開源技術(shù)峰會
國家信息中心與摩爾線程達(dá)成戰(zhàn)略合作
摩爾線程發(fā)布Torch-MUSA v2.1.1版本
【「DeepSeek 核心技術(shù)揭秘」閱讀體驗(yàn)】書籍介紹+第一章讀后心得
摩爾線程率先支持騰訊混元-A13B模型
【書籍評測活動NO.62】一本書讀懂 DeepSeek 全家桶核心技術(shù):DeepSeek 核心技術(shù)揭秘
摩爾線程GPU率先支持Qwen3全系列模型
摩爾線程GPU成功適配Deepseek-V3-0324大模型
摩爾線程發(fā)布云電腦驅(qū)動MT vGPU 2.7.0
摩爾線程GPU原生FP8計(jì)算助力AI訓(xùn)練
摩爾線程支持阿里云通義千問QwQ-32B開源模型
摩爾線程支持DeepSeek開源通信庫DeepEP和并行算法DualPipe
評論