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é)點內(nèi)通信
訓(xùn)練及推理預(yù)填充階段的高吞吐量計算核心
推理解碼階段的低延遲計算核心
原生支持 FP8 數(shù)據(jù)分發(fā)
靈活控制 GPU 資源,實現(xiàn)計算與通信的高效重疊
DualPipe是DeepSeek-V3提出的雙向流水線并行算法,通過前向計算與后向計算階段的計算與通信完全重疊,顯著減少了“流水線氣泡”(設(shè)備空閑等待)。與傳統(tǒng)流水線并行相比,DualPipe 采用雙向數(shù)據(jù)流設(shè)計,使數(shù)據(jù)從兩端相向處理,大幅提升了資源利用率與訓(xùn)練效率。
摩爾線程依托深度學(xué)習(xí)框架Torch-MUSA(已開源)和MUSA軟件棧全方位的兼容性,實現(xiàn)了對DualPipe這一算法的高效支持。MT-DualPipe可以完整接入摩爾線程 MT-Megatron框架和MT-TransformerEngine框架(即將開源),實現(xiàn)DeepSeek V3訓(xùn)練流程的完整復(fù)現(xiàn)。
此外,MT-DualPipe結(jié)合MT-Megatron可以實現(xiàn)完整DeepSeek V3模型的MLP-FFN分離以及DW-DG分離,進一步降低氣泡占比,優(yōu)化通信效率。同時,MT-DualPipe與MT-TranformerEngine和MT-DeepEP的結(jié)合,可以利用MT-DeepEP和異步通信引擎實現(xiàn)更高效的通信掩蓋,大大降低對計算資源的損耗。
DeepEP和DualPipe的第一時間適配,充分驗證了摩爾線程MUSA架構(gòu)和全功能GPU在生態(tài)兼容與快速適配上的強大優(yōu)勢。我們誠摯歡迎廣大開發(fā)者下載體驗,并期待您的反饋與建議。
關(guān)于摩爾線程
摩爾線程成立于2020年10月,以全功能GPU為核心,致力于向全球提供加速計算的基礎(chǔ)設(shè)施和一站式解決方案,為各行各業(yè)的數(shù)智化轉(zhuǎn)型提供強大的AI計算支持。
我們的目標是成為具備國際競爭力的GPU領(lǐng)軍企業(yè),為融合人工智能和數(shù)字孿生的數(shù)智世界打造先進的加速計算平臺。我們的愿景是為美好世界加速。
-
開源
+關(guān)注
關(guān)注
3文章
4346瀏覽量
46442 -
摩爾線程
+關(guān)注
關(guān)注
2文章
285瀏覽量
6662 -
DeepSeek
+關(guān)注
關(guān)注
2文章
839瀏覽量
3400
原文標題:DeepSeek開源|摩爾線程完成對DeepEP和DualPipe的支持
文章出處:【微信號:moorethreads,微信公眾號:摩爾線程】歡迎添加關(guān)注!文章轉(zhuǎn)載請注明出處。
發(fā)布評論請先 登錄
摩爾線程正式開源MuJoCo Warp MUSA
摩爾線程斬獲6.6億元合同訂單
尋找對RISCV眾核并行計算感興趣的伙伴、朋友
尋找對RISCV眾核并行計算感興趣的伙伴
摩爾線程MTT S5000全面適配Qwen3.5三款新模型
摩爾線程正式開源TileLang-MUSA項目
摩爾線程高開468% 中一簽賺27萬 國產(chǎn)GPU第一股摩爾線程高開468%
摩爾線程支持DeepSeek開源通信庫DeepEP和并行算法DualPipe
評論