IT之家 5 月 7 日消息，騰訊技術(shù)團(tuán)隊(duì)針對 DeepSeek 開源的 DeepEP 通信框架進(jìn)行深度優(yōu)化，使其在多種網(wǎng)絡(luò)環(huán)境下均實(shí)現(xiàn)顯著性能提升。

經(jīng)測試，優(yōu)化后的通信框架性能在 RoCE 網(wǎng)絡(luò)環(huán)境提升 100%，IB 網(wǎng)絡(luò)環(huán)境提升 30%，為企業(yè)開展 AI 大模型訓(xùn)練提供更高效的解決方案。

該技術(shù)方案獲得了 DeepSeek 公開致謝，稱這是一次“huge speedup”代碼貢獻(xiàn)。

打開網(wǎng)易新聞 查看精彩圖片

IT之家注：DeepEP 是一個為專家混合（MoE）和專家并行（EP）量身定制的通信庫。它提供高吞吐量和低延遲的全對全 GPU 內(nèi)核，這些內(nèi)核也被稱為 MoE 派發(fā)和組合。該庫還支持低精度運(yùn)算，包括 FP8。

據(jù)悉，騰訊星脈網(wǎng)絡(luò)團(tuán)隊(duì)在 DeepEP 開源后便展開技術(shù)攻關(guān)，發(fā)現(xiàn)兩大關(guān)鍵瓶頸：一是對于雙端口網(wǎng)卡帶寬利用率不足，二是 CPU 控制面交互存在時延。

在騰訊的技術(shù)優(yōu)化下，DeepEP 不僅在 RoCE 網(wǎng)絡(luò)實(shí)現(xiàn)性能翻倍，反哺到 IB（InfiniBand）網(wǎng)絡(luò)時更使原有通信效率再提升 30%。

目前，該技術(shù)已全面開源，并成功應(yīng)用于騰訊混元大模型等項(xiàng)目的訓(xùn)練推理，在騰訊星脈與 H20 服務(wù)器構(gòu)建的環(huán)境中，這套方案展現(xiàn)出出色的通用性。