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提出的雙向流水線并行算法,通過前向計算與后向計算階段的計算與通信完全重疊,顯著減少了“流水線氣泡”(設備空閑等待)。與傳統流水線并行相比,DualPipe 采用雙向數據流設計,使數據從兩端相向處理,大幅提升了資源利用率與訓練效率。
摩爾線程依托深度學習框架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和異步通信引擎實現更高效的通信掩蓋,大大降低對計算資源的損耗。
DeepEP和DualPipe的第一時間適配,充分驗證了摩爾線程MUSA架構和全功能GPU在生態兼容與快速適配上的強大優勢。我們誠摯歡迎廣大開發者下載體驗,并期待您的反饋與建議。
關于摩爾線程
摩爾線程成立于2020年10月,以全功能GPU為核心,致力于向全球提供加速計算的基礎設施和一站式解決方案,為各行各業的數智化轉型提供強大的AI計算支持。
我們的目標是成為具備國際競爭力的GPU領軍企業,為融合人工智能和數字孿生的數智世界打造先進的加速計算平臺。我們的愿景是為美好世界加速。
-
開源
+關注
關注
3文章
4173瀏覽量
45933 -
摩爾線程
+關注
關注
2文章
276瀏覽量
6405 -
DeepSeek
+關注
關注
2文章
833瀏覽量
3194
原文標題:DeepSeek開源|摩爾線程完成對DeepEP和DualPipe的支持
文章出處:【微信號:moorethreads,微信公眾號:摩爾線程】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
摩爾線程正式開源TileLang-MUSA項目
摩爾線程發布Torch-MUSA v2.7.0版本
【「DeepSeek 核心技術揭秘」閱讀體驗】書籍介紹+第一章讀后心得
【書籍評測活動NO.62】一本書讀懂 DeepSeek 全家桶核心技術:DeepSeek 核心技術揭秘
摩爾線程GPU成功適配Deepseek-V3-0324大模型
摩爾線程GPU原生FP8計算助力AI訓練
摩爾線程全面支持DeepSeek開源周成果
摩爾線程完成DeepSeek開源庫FlashMLA和DeepGEMM適配
DeepSeek宣布開源DeepGEMM
DeepSeek扔的第二枚開源王炸是什么,跟第一枚有何不同?強烈引發外媒熱議
沐曦GPU跑通DeepSeek開源代碼庫FlashMLA
摩爾線程圖形顯卡MTT S80實現DeepSeek模型部署
摩爾線程支持DeepSeek開源通信庫DeepEP和并行算法DualPipe
評論