DeepSeek開源第四彈:并行優(yōu)化策略3項齊開

2月27日消息,DeepSeek開源周第四日,DeepSeek開源了并行優(yōu)化策略(Optimized Parallelism Strategies),一次開源了3項:

DualPipe:一種用于V3/R1模型訓(xùn)練中實現(xiàn)計算與通信重疊的雙向流水線并行算法

EPLB:一個針對V3/R1的專家并行負載均衡工具

Profile-data:訓(xùn)練和推理框架的分析數(shù)據(jù)

GitHub地址:

https://github.com/deepseek-ai/DualPipe

https://github.com/deepseek-ai/eplb

https://github.com/deepseek-ai/profile-data

一、DualPipe

DualPipe是DeepSeek-V3技術(shù)報告中提出的一種創(chuàng)新雙向流水線并行算法。

它能夠?qū)崿F(xiàn)前向與后向計算和通信階段的完全重疊,同時有效減少流水線氣泡(空閑時間)。

DeepSeek展示了在8個流水線并行階段和20個micro-batches情況下,DualPipe在兩個方向上的調(diào)度示例。(來自DeepSeek-V3技術(shù)報告)

反向方向的微批次與前向方向?qū)ΨQ,為了簡化圖示,這里省略了反向方向的批次 ID。圖中由同一個黑色邊框包圍的兩個單元格具有相互重疊的計算和通信。

流水線氣泡與內(nèi)存使用比較

表中,

(免責(zé)聲明:本網(wǎng)站內(nèi)容主要來自原創(chuàng)、合作伙伴供稿和第三方自媒體作者投稿,凡在本網(wǎng)站出現(xiàn)的信息,均僅供參考。本網(wǎng)站將盡力確保所提供信息的準(zhǔn)確性及可靠性,但不保證有關(guān)資料的準(zhǔn)確性及可靠性,讀者在使用前請進一步核實,并對任何自主決定的行為負責(zé)。本網(wǎng)站對有關(guān)資料所引致的錯誤、不確或遺漏,概不負任何法律責(zé)任。
任何單位或個人認為本網(wǎng)站中的網(wǎng)頁或鏈接內(nèi)容可能涉嫌侵犯其知識產(chǎn)權(quán)或存在不實內(nèi)容時,應(yīng)及時向本網(wǎng)站提出書面權(quán)利通知或不實情況說明,并提供身份證明、權(quán)屬證明及詳細侵權(quán)或不實情況證明。本網(wǎng)站在收到上述法律文件后,將會依法盡快聯(lián)系相關(guān)文章源頭核實,溝通刪除相關(guān)內(nèi)容或斷開相關(guān)鏈接。 )

贊助商
2025-02-27
DeepSeek開源第四彈:并行優(yōu)化策略3項齊開
DeepSeek展示了在8個流水線并行階段和20個micro-batches情況下,DualPipe在兩個方向上的調(diào)度示例。反向方向的微批次與前向方向?qū)ΨQ,為了

長按掃碼 閱讀全文