5月7日,記者獲悉,騰訊技術(shù)團隊針對DeepSeek開源的DeepEP通信框架進行深度優(yōu)化,使其在多種網(wǎng)絡環(huán)境下均實現(xiàn)顯著性能提升。經(jīng)測試,優(yōu)化后的通信框架性能在RoCE網(wǎng)絡環(huán)境提升100%,IB網(wǎng)絡環(huán)境提升30%,為企業(yè)開展AI大模型訓練提供更高效的解決方案。相關(guān)技術(shù)方案獲得了DeepSeek公開致謝,稱這是一次“huge speedup”代碼貢獻。
自今年2月DeepSeek開源包括DeepEP在內(nèi)的五大代碼庫以來,該團隊便向業(yè)界展示了如何利用有限的硬件資源實現(xiàn)接近萬卡集群的性能。在這些技術(shù)中,DeepEP憑借突破性的方法提升了300%的通信效率,成功解決了MoE架構(gòu)大模型對英偉達NCCL的依賴問題。
但該技術(shù)在成本較低、適用面更廣的RoCE網(wǎng)絡環(huán)境中表現(xiàn)不佳,限制了其在更廣泛場景的應用。這一痛點引發(fā)了開源社區(qū)的持續(xù)討論。
騰訊星脈網(wǎng)絡團隊基于在RoCE網(wǎng)絡領(lǐng)域的深厚積累,在DeepEP開源后便展開技術(shù)攻關(guān),發(fā)現(xiàn)兩大關(guān)鍵瓶頸:一是對于雙端口網(wǎng)卡帶寬利用率不足,二是CPU控制面交互存在時延。
騰訊在RoCE網(wǎng)絡優(yōu)化方面的突破,首先體現(xiàn)在帶寬分配的智能化,通過拓撲感知的多QP建鏈技術(shù),智能分配數(shù)據(jù)流,優(yōu)化了雙端口網(wǎng)卡的帶寬利用率,確保每條數(shù)據(jù)通道都能得到充分使用。此舉有效避免了帶寬浪費,為RoCE網(wǎng)絡的性能提升提供了有力支撐。
其次,騰訊還著力解決了GPU通信中的CPU控制瓶頸問題。通過基于IBGDA技術(shù)的優(yōu)化,騰訊使得“控制面”場景的操作也可以繞過CPU的“中轉(zhuǎn)”,進一步降低了延遲和能耗,提升了整體通信效率。
同時,GPU直接“對話”時存在的傳輸順序混亂難題,騰訊提出了“QP內(nèi)時序鎖”機制,使得多個GPU間的數(shù)據(jù)傳輸能夠精準、按順序完成,即使同時處理1000多個數(shù)據(jù)傳輸任務,DeepEP也能自動理順先后順序。
在騰訊的技術(shù)優(yōu)化下,DeepEP不僅在RoCE網(wǎng)絡實現(xiàn)性能翻倍,反哺到IB(InfiniBand)網(wǎng)絡時更使原有通信效率再提升30%。
(騰訊工程師在不同節(jié)點服務器上的測試數(shù)據(jù))
目前該技術(shù)已全面開源,并成功應用于騰訊混元大模型等項目的訓練推理,在騰訊星脈與H20服務器構(gòu)建的高性能環(huán)境中,這套方案同樣展現(xiàn)出卓越的通用性。
免責聲明:此文內(nèi)容為第三方自媒體作者發(fā)布的觀察或評論性文章,所有文字和圖片版權(quán)歸作者所有,且僅代表作者個人觀點,與極客網(wǎng)無關(guān)。文章僅供讀者參考,并請自行核實相關(guān)內(nèi)容。投訴郵箱:editor@fromgeek.com。
免責聲明:本網(wǎng)站內(nèi)容主要來自原創(chuàng)、合作伙伴供稿和第三方自媒體作者投稿,凡在本網(wǎng)站出現(xiàn)的信息,均僅供參考。本網(wǎng)站將盡力確保所提供信息的準確性及可靠性,但不保證有關(guān)資料的準確性及可靠性,讀者在使用前請進一步核實,并對任何自主決定的行為負責。本網(wǎng)站對有關(guān)資料所引致的錯誤、不確或遺漏,概不負任何法律責任。任何單位或個人認為本網(wǎng)站中的網(wǎng)頁或鏈接內(nèi)容可能涉嫌侵犯其知識產(chǎn)權(quán)或存在不實內(nèi)容時,應及時向本網(wǎng)站提出書面權(quán)利通知或不實情況說明,并提供身份證明、權(quán)屬證明及詳細侵權(quán)或不實情況證明。本網(wǎng)站在收到上述法律文件后,將會依法盡快聯(lián)系相關(guān)文章源頭核實,溝通刪除相關(guān)內(nèi)容或斷開相關(guān)鏈接。