摩爾線程助攻深度尋源開源周:打造“全家桶”助力開源生態(tài)繁榮

摩爾線程助攻深度尋源開源周:打造“全家桶”助力開源生態(tài)繁榮

自 IT 之家 3 月 2 日消息,DeepSeek 開源周已正式收官,這一周對(duì)于摩爾線程智能科技(北京)股份有限公司來(lái)說,無(wú)疑是一次重大的里程碑。在短短時(shí)間內(nèi),摩爾線程成功實(shí)現(xiàn)對(duì) DeepSeek 各個(gè)開源項(xiàng)目的全面支持,涵蓋了 FlashMLA、DeepEP、DeepGEMM、DualPipe 以及 Fire-Flyer 文件系統(tǒng)(3FS)。這一切的背后,是摩爾線程對(duì)開源生態(tài)的深度參與和積極推動(dòng)。

FlashMLA,一款高效的 MLA(Multi-Head Latent Attention)推理內(nèi)核開源倉(cāng)庫(kù),旨在加速 MLA 機(jī)制的計(jì)算,特別適用于 DeepSeek 系列模型。摩爾線程基于全新 MUSA Compute Capability 3.1 計(jì)算架構(gòu),可提供原生 FP8 計(jì)算能力。同時(shí),升級(jí)了高性能線性代數(shù)模板庫(kù) MUTLASS,從而快速支持了 FlashMLA。借助 MUTLASS 0.2.0,摩爾線程發(fā)布開源倉(cāng)庫(kù) MT-FlashMLA,能夠快速對(duì) DeepSeek FlashMLA 進(jìn)行兼容部署。

DeepEP,一個(gè)用于 MoE(混合專家)模型訓(xùn)練和推理的開源 EP(expert parallelism,專家并行)通信庫(kù)。DeepEP 主要適用于大模型訓(xùn)練,特別是需要 EP 的集群訓(xùn)練。摩爾線程基于 MUSA Compute Capability 3.1 全功能 GPU,第一時(shí)間適配了 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ì)算與通信的高效重疊等特點(diǎn),無(wú)疑將為 DeepEP 的廣泛應(yīng)用提供了有力支持。

DeepGEMM,一個(gè)支持密集矩陣與混合專家(MoE)矩陣乘法的 FP8 GEMM 庫(kù)。摩爾線程在全新 GPU 架構(gòu)上優(yōu)化實(shí)現(xiàn)了 FP8 矩陣乘法,從而支持 DeepGEMM 的相應(yīng)功能。與此同時(shí),摩爾線程依托深度學(xué)習(xí)框架 Torch-MUSA 和 MUSA 軟件棧全方位的兼容性,實(shí)現(xiàn)了對(duì) DeepGEMM 的高效支持。

DualPipe,DeepSeek-V3 提出的雙向流水線并行算法,通過前向計(jì)算與后向計(jì)算階段的計(jì)算與通信完全重疊,顯著減少了“流水線氣泡”(設(shè)備空閑等待)。與傳統(tǒng)流水線并行相比,DualPipe 采用雙向數(shù)據(jù)流設(shè)計(jì),使數(shù)據(jù)從兩端相向處理,大幅提升了資源利用率與訓(xùn)練效率。摩爾線程在 DualPipe 的支持上表現(xiàn)出色,依托深度學(xué)習(xí)框架 Torch-MUSA 和 MUSA 軟件棧全方位的兼容性,實(shí)現(xiàn)了對(duì) DualPipe 這一算法的高效支持。

至于 Fire-Flyer 文件系統(tǒng)(3FS),這是一種利用現(xiàn)代 SSD 和 RDMA 網(wǎng)絡(luò)的全部帶寬的并行文件系統(tǒng),可以把固態(tài)硬盤的帶寬性能利用到極致。在 V3 和 R1 的訓(xùn)練與推理過程中,3FS 作為關(guān)鍵支撐,以應(yīng)對(duì) AI 訓(xùn)練和推理工作負(fù)載的挑戰(zhàn)。為了高效支持 3FS,摩爾線程在一天內(nèi)迅速完成了高性能分布式文件系統(tǒng) 3FS 的搭建,并高效開發(fā)了存儲(chǔ)插件,成功實(shí)現(xiàn)與夸娥智算集群的無(wú)縫集成。

總的來(lái)說,摩爾線程以實(shí)際行動(dòng)參與和推動(dòng) DeepSeek 開源周活動(dòng),不僅提供了全面的技術(shù)支持,還打造了一系列開源項(xiàng)目——“全家桶”,涵蓋了多個(gè)關(guān)鍵領(lǐng)域。這些開源項(xiàng)目不僅展示了摩爾線程的技術(shù)實(shí)力和開放精神,也以實(shí)際行動(dòng)助力開源生態(tài)繁榮。在未來(lái)的日子里,我們期待看到更多這樣的開源項(xiàng)目和活動(dòng),共同推動(dòng)科技的發(fā)展和進(jìn)步。

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

贊助商
2025-03-02
摩爾線程助攻深度尋源開源周:打造“全家桶”助力開源生態(tài)繁榮
摩爾線程積極參與并推動(dòng)深度尋源開源周活動(dòng),提供全面的技術(shù)支持和開源項(xiàng)目,助力開源生態(tài)繁榮。

長(zhǎng)按掃碼 閱讀全文