幾個月前在調一套8卡訓練集群的性能時我發(fā)現(xiàn)一個很反常的現(xiàn)象把某個圖像分類模型的全局batch從2調到4之后吞吐量幾乎沒漲nvidia-smi里的GPU利用率卻掉了一截。后來把訓練步耗時拆開看梯度同步占了將近三成。也就是從那天開始我認真去扒了NCCL底層到底在做什么也從NCCL Ring LL一路走到自己用PCIe IPC實現(xiàn)了一個定制版的AllReduce。這篇文章不是什么高深理論課而是把我從“只會調NCCL環(huán)境變量”到“自己動手寫通信原語”的完整過程寫出來。內容包括Ring算法的搬運模型、LL協(xié)議到底優(yōu)化了什么、為什么單機PCIe場景下NCCL未必最優(yōu)、我自己實現(xiàn)的PCIe IPC AllReduce怎么設計、以及實測中兩種方案的差異和踩坑記錄。適合正在做單機多卡訓練調優(yōu)、對分布式通信庫底層感興趣、或者想自己定制集合通信邏輯的開發(fā)者。1. 先厘清起點Ring LL在單機多卡里到底贏了什么1.1 AllReduce為什么是訓練里的“熱車道”數(shù)據(jù)并行訓練里每張卡算完自己的batch之后梯度必須做一次全局歸約保證所有卡拿到一致的梯度才能繼續(xù)下一步。這個操作就是AllReduce。你可以把它理解為一場考試后全班對答案的環(huán)節(jié)——每個人手上有一份自己的答案必須兩兩核對、合并出最終標準答案然后所有人再抄回去。每次訓練step都會執(zhí)行一次所以它不是在后臺跑一次的離線任務而是實打實占在關鍵路徑上的“熱車道”。在單機8卡這種場景里AllReduce的耗時直接影響訓練吞吐。模型越大、batch越大梯度張量越大這個時間就越刺眼。很多同學遇到訓練慢的第一反應是改學習率、換網絡結構但真正的問題可能出在通信這塊。1.2 Ring算法為什么是環(huán)形接力賽NCCL最廣為人知的AllReduce實現(xiàn)是Ring AllReduce。它的核心思想很有意思把每個rank的梯度張量切成N份N是通信域里的rank數(shù)量然后讓數(shù)據(jù)在一個邏輯環(huán)上接力傳遞。假設有4張卡每個rank的梯度張量被切成4塊。第一步叫ReduceScatter階段每個rank把自己的一部分數(shù)據(jù)塊傳給鄰居同時接收上一個鄰居傳來的數(shù)據(jù)塊并對兩塊做求和。經過N-1輪傳遞后每個rank手里恰好握著一塊“所有rank在對應位置上的歸約結果”。第二步叫AllGather階段每個rank把自己手里的最終結果塊沿著環(huán)再傳一遍所有人都收到完整的歸約結果。這個過程的通信量公式是2×M×(N-1)/N其中M是消息總大小。當N8時每個rank發(fā)出的數(shù)據(jù)是總數(shù)據(jù)量的1.75倍理論上已經非常接近下界。Ring之所以成為默認算法就是因為它對帶寬的利用率高而且不依賴某個中心節(jié)點做匯聚不容易出現(xiàn)單點瓶頸。1.3 LL協(xié)議的提速本質把“等確認”從關鍵路徑上拿掉NCCL在Ring基礎上還有一個關鍵變體叫LL即Low Latency低延遲協(xié)議。很多人以為LL只是把消息切得更小其實它真正的精髓在于改變了接收端等待數(shù)據(jù)的方式。普通協(xié)議里發(fā)送方發(fā)完數(shù)據(jù)后要等接收方確認收到才能釋放buffer繼續(xù)下一輪。這個“等待確認”的時間長則幾十微秒短則幾微秒但對于小消息來說這個開銷可能比數(shù)據(jù)搬運本身還大。LL協(xié)議的做法是把數(shù)據(jù)切成長度很小的塊每塊數(shù)據(jù)后面緊跟一個flag標記位。發(fā)送方寫完數(shù)據(jù)后更新flag接收方不再等待任何顯式的完成通知而是不停自旋輪詢相應的flag地址——一旦發(fā)現(xiàn)flag變化就立刻認為數(shù)據(jù)已經到位可以開算。這個過程被稱為“flag-based同步”。打個比方普通協(xié)議相當于快遞送完包裹后要等收件人簽字回執(zhí)快遞員才送下一件。LL協(xié)議則是快遞員直接把包裹扔到門口然后通過門鈴按鈕通知收件人自己繼續(xù)送下一件收件人聽到門鈴就出來取。省掉的回執(zhí)時間在小消息場景下非常可觀這也是為什么NCCL在處理小梯度消息時LL往往比其他協(xié)議明顯更快。但這里有個前置條件flag輪詢必須非常頻繁而且是忙等不是休眠喚醒。這就意味著收發(fā)雙方都需要在同一個物理設備上能看到同一塊內存否則輪詢本身就會引入延遲。而這一點恰恰是單機PCIe場景的天然優(yōu)勢也是我后續(xù)自研方案的基礎。2. 我為什么覺得NCCL的Ring LL還不夠把延遲賬本拆開算2.1 賬本里有三筆錢協(xié)議狀態(tài)機、flag輪詢、內核同步NCCL的Ring LL已經很快了但在我的場景里還是不夠。原因不是它慢而是我看完它的工作機制后發(fā)現(xiàn)里面有些開銷對我來說不是必須的。第一筆開銷在協(xié)議狀態(tài)機。NCCL內部有完整的狀態(tài)管理連接建立、buffer分配、傳輸進度跟蹤、錯誤重試。這些機制在多機環(huán)境或者大規(guī)模集群里是保命用的但在單機8卡、拓撲固定、進程生命周期可控的場景下大部分狀態(tài)機邏輯其實是在陪跑。每一次通信都要過一遍狀態(tài)機哪怕狀態(tài)沒有變化也會在關鍵路徑上增加若干次分支判斷和內存訪問。第二筆開銷是flag輪詢的粒度。LL協(xié)議已經做得很快但它是通用設計需要同時適配NVLink、PCIe、網卡等多種硬件路徑。在PCIe上它選擇的flag更新方式和輪詢地址布局未必是針對當前物理鏈路最優(yōu)的選擇。第三筆開銷來自內核同步。NCCL在通信結束或者某些階段切換時仍會涉及CUDA事件、內核launch等同步點。這些同步點保證正確性但同時會打斷GPU流水線。在我這種每步都跑AllReduce的場景里流水線被打斷的代價比想象中大。2.2 PCIe拓撲才是單機多卡的真瓶頸單機多卡環(huán)境下最容易被忽略的是PCIe拓撲本身。8張卡插在一臺機器上走的是PCIe switch或多組PCIe通道。不同卡之間的數(shù)據(jù)通路其實是有差異的有的卡對之間是直連通道帶寬高延遲低有的卡對之間要繞兩個switch帶寬受限。NCCL的Ring算法有一個隱含假設環(huán)上相鄰節(jié)點的通信代價大致均勻。但在PCIe拓撲下這個假設經常不成立。如果環(huán)上恰好安排了兩張卡跨越帶寬很窄的鏈路整個Ring的帶寬會被這條鏈路拖住。這也是為什么NCCL在建立通信域時會做拓撲探測和路徑排序的嘗試但通用庫必須兼容各種奇怪的拓撲做不到為某一臺具體機器定制最優(yōu)環(huán)序。我當時拿工具測了一下8卡兩兩之間的實際帶寬矩陣發(fā)現(xiàn)最差鏈路的雙向帶寬只有最好鏈路的六成左右。這直接意味著Ring里的“木桶效應”非常突出。而我后續(xù)自研的IPC方案恰恰可以在數(shù)據(jù)切分和流量分配上避開這些劣勢鏈路。2.3 更現(xiàn)實的需求可控的通信步調與自定義歸約第三個理由比較實際我想把通信的步調控制在自己手里。NCCL太“聰明”了它會根據(jù)消息大小自動選擇算法、協(xié)議、buffer策略。這在多數(shù)時候是好事但當你需要做性能實驗、對比不同切分策略、或者把通信和計算重疊在一起調度時這種自動化反而讓人抓狂。我記得有次想固定使用Ring LL協(xié)議來做對照試驗結果NCCL根據(jù)消息尺寸自己切到了別的協(xié)議實驗數(shù)據(jù)直接沒法對齊。當時我就想如果在單機PCIe場景下自己實現(xiàn)一個行為完全可預測的AllReduce把所有開關暴露出來是不是就能解決這類問題?；仡^看這個想法有點“造輪子”的沖動但也正是這種沖動讓我在后面幾周里對通信底層的理解上了一個臺階。3. PCIe IPC AllReduce的總體設計讓數(shù)據(jù)搬運變成內存拷貝3.1 設計目標在PCIe帶寬邊界附近做文章先說明這套方案的適用邊界單機多卡所有GPU都能通過PCIe互相訪問對方內存進程分布在同一個節(jié)點上。不適用于多機場景不適用于GPU直連NVLink全覆蓋的拓撲。在這個前提下我給自研方案定了三個目標。第一把通信開銷壓到接近純內存拷貝的水平。PCIe上的GPU對等訪問本質上是把數(shù)據(jù)從一張卡的顯存搬到另一張卡的顯存中間走PCIe總線。如果能繞開NCCL的協(xié)議層直接把“數(shù)據(jù)搬運”這件事做成“共享內存讀寫”就能最大程度貼近硬件物理極限。第二同步機制要做到輕量。我選擇在GPU側用flag輪詢做同步避免CPU參與每次數(shù)據(jù)到達的確認。這和NCCL LL的思路一致但實現(xiàn)得更“裸”只保留最基本的生產者-消費者標記。第三拓撲感知的流量調度。我提前測出8卡兩兩之間的有效帶寬矩陣并據(jù)此安排通信環(huán)的順序和單次傳輸?shù)膲K大小而不是機械地按rank編號繞圈。3.2 通信模式拆解ReduceScatter AllGather我的實現(xiàn)依然沿用ReduceScatter加AllGather的經典兩步拆解沒有另辟蹊徑。原因很簡單這兩步本身已經接近通信量下界換別的模式比如二叉樹在單機PCIe場景反而會因為鏈路復用率低而更慢。具體來說每個rank把自己的梯度張量按通信域大小切成N塊。ReduceScatter階段沿著邏輯環(huán)做N-1輪收發(fā)與求和結束后每個rank持有一塊全局歸約結果。AllGather階段再把這N塊結果沿環(huán)廣播一輪最終所有rank拿到完整歸約張量。這里我做的第一個優(yōu)化是消息切分的粒度不固定而是根據(jù)帶寬矩陣動態(tài)決定。測試發(fā)現(xiàn)某些相鄰節(jié)點鏈路帶寬高就讓它多傳一些數(shù)據(jù)鏈路窄的節(jié)點對就少傳。這打破了Ring原本“每輪傳等量數(shù)據(jù)”的均勻假設雖然實現(xiàn)復雜度上升了但帶寬利用率明顯改善。3.3 共享內存布局與IPC映射流程整個方案的根基是CUDA的IPC能力也就是讓多個進程共享同一塊顯存。流程并不復雜但每一步都不能錯。首先由一個主進程分配一塊GPU顯存作為通信buffer池。每個rank在這塊顯存里劃分出兩個區(qū)域data區(qū)存放實際的梯度數(shù)據(jù)分塊flag區(qū)存放數(shù)據(jù)塊對應的完成標記。然后主進程通過CUDA IPC API拿到這塊顯存的IPC handle分發(fā)給其他所有rank進程。其他進程拿到handle后在自己的地址空間里打開映射獲得同一塊物理顯存的不同虛擬地址。這里的關鍵點是IPC handle不是普通的指針不能直接跨進程傳遞。它是CUDA內部用于標識物理顯存分配的句柄必須配合打開操作才能變成可用的映射地址。在我的實現(xiàn)里主進程通過一個共享文件或者Unix domain socket把handle廣播出去其他進程再各自調用cuIpcOpenMemHandle完成映射。映射完成后所有rank都擁有了讀寫通信buffer的能力。這一步到位之后剩下的AllReduce邏輯就變成了純粹的內存讀寫加flag輪詢不再需要任何庫級別的通信API。3.4 一次AllReduce的完整時間線用偽代碼來描述一次完整的AllReduce流程會清晰很多。這里以4個rank為例。// 偽代碼PCIe IPC AllReduce 單步流程 // chunk為數(shù)據(jù)塊編號step為Ring輪次 for (step 0; step numRanks - 1; step) { sendRank (myRank step) % numRanks; recvRank (myRank - step numRanks) % numRanks; for (chunk 0; chunk numChunks; chunk) { if (chunk_is_local_responsibility(myRank, chunk, step)) { // 本地歸約本地數(shù)據(jù)塊 收到的數(shù)據(jù)塊 reduce_local_chunk(chunk); } // 將數(shù)據(jù)塊寫入send buffer對應位置 write_data_to_send_slot(sendRank, chunk); // 寫入完成后更新flag update_flag(sendRank, chunk); // 自旋輪詢recv slot的flag while (read_flag(recvRank, chunk) ! expected_value) { } read_data_from_recv_slot(recvRank, chunk); } }大致時間線是這樣的每個rank在step開始時先嘗試把本地已經歸約過的數(shù)據(jù)塊寫入自己的發(fā)送槽位并更新flag同時自旋等待相鄰rank的接收槽位flag變化。flag一變立刻讀取數(shù)據(jù)參與歸約。這個流程沒有全局屏障完全靠生產者和消費者之間的flag協(xié)作推進所以端到端延遲主要取決于最慢節(jié)點的寫入時延加PCIe傳輸時延。我第一次跑通時有點驚訝沒有用任何NCCL API也沒走網卡純靠PCIe對等映射和flag自旋就把1GB級別的梯度全量同步做完了。從那一刻起我才真正理解庫能給你的只是封裝物理鏈路本身能跑多快取決于你愿不愿意花時間去貼近它。4. 落地過程中最容易被忽略的五個細節(jié)4.1 IPC handle不能從任意指針拿先找到分配塊基址第一個讓我翻車的點在獲取IPC handle上。我一開始很自然地對一個從buffer池里偏移出來的指針直接調用了IPC handle獲取API結果返回錯誤。查了文檔才明白IPC handle是作用在整個分配塊上的不是作用在任意子指針上的。你必須先用cuMemGetAddressRange把指針所在的分配塊基址和大小找出來再用基址去獲取IPC handle。這個細節(jié)在實際編碼時很容易漏掉尤其是當通信buffer池里劃分了data區(qū)和flag區(qū)、并且每個rank都拿到不同子區(qū)時。如果每個rank都試圖拿自己子區(qū)的IPC handle不僅會出錯還會在權限管理上制造混亂。我的做法是主進程一次性把整個buffer池的IPC handle拿出來分發(fā)給所有rank各rank再在自己的映射里計算子區(qū)偏移。這樣既合法又避免了重復獲取handle的負擔。4.2 同步不能只靠volatile內存柵欄要放在正確位置flag同步最隱蔽的坑是內存序問題。很多第一次寫GPU flag輪詢的人會寫類似這樣的代碼發(fā)送方先寫data再寫flag接收方輪詢flag看到變化后讀data。聽起來天衣無縫但實際在GPU和PCIe的組合下data的寫入不一定在flag寫入被對方看到之前完成。這里涉及的是“可見性”順序光靠volatile關鍵字指示編譯器不要優(yōu)化讀取不夠硬件層仍然可能重排寫操作。生產者必須在寫完data后、寫flag前插入內存柵欄指令確保data的寫入已經沖刷到目標可見層級消費者讀到flag變化后、讀data前也要插入內存柵欄確保后續(xù)讀取能看到data的最新值。我在最初版本里把內存柵欄放在了錯誤的位置——放在寫flag之后。結果發(fā)現(xiàn)接收方偶爾讀到的是舊數(shù)據(jù)排查了很久才知道問題出在柵欄順序上。后來改成標準姿勢寫datathreadfence寫flag讀flag變化后threadfence讀data。這個組合看起來簡單但少一個fence整個方案就是錯的。4.3 數(shù)據(jù)切分要匹配PCIe鏈路的實際拓撲前面說過Ring假設相鄰鏈路均勻但PCIe拓撲并不均勻。我的自研方案做了一個很關鍵的調整在建立通信域時先跑一遍小規(guī)模帶寬探測把8卡之間兩兩的實際雙向帶寬矩陣算出來然后用貪心算法構造一條“相對均勻”的環(huán)。具體做法是從任意一張卡出發(fā)每次選擇與當前節(jié)點帶寬最大的未訪問節(jié)點作為環(huán)上下一個鄰居直到所有節(jié)點都被納入。這個環(huán)不一定是最短路徑但能保證環(huán)上每條邊的帶寬不會出現(xiàn)劇烈低谷。然后在ReduceScatter階段根據(jù)每條邊的真實帶寬比例來分配每個數(shù)據(jù)塊應該傳輸?shù)目傋止?jié)數(shù)帶寬高的邊承載更大的塊。這個優(yōu)化做進來之后整體同步耗時下降了大約一成多。雖然不算驚人但在一遍遍跑的訓練循環(huán)里這個比例帶來的吞吐提升是實打實的。4.4 flag區(qū)和data區(qū)不能共享同一個cache lineGPU上的緩存行大小通常是128字節(jié)。如果你把flag和它對應的data放在同一個緩存行里接收方輪詢flag時硬件會把整個緩存行拉取到緩存里同時發(fā)送方寫入data時也會更新整個緩存行。兩者互相干擾會造成嚴重的false sharing表現(xiàn)為輪詢延遲飆升和帶寬下降。我的做法是在buffer池里把flag區(qū)單獨劃分出來并且讓每個flag和相鄰data之間保留足夠的填充間隔確保flag不會和任何data共享緩存行。實測對比過有填充和無填充的版本前者在200MB以上大消息場景下帶寬比后者高出接近10%。這個優(yōu)化幾乎零成本強烈建議所有做類似方案的人都加上。4.5 進程生命周期管理映射釋放比建立更重要IPC映射的建立很爽但釋放是另一回事。如果某個rank進程異常退出或者主進程提前釋放了buffer池其他rank仍持有打開狀態(tài)的IPC映射就會導致顯存資源無法回收甚至讓整個CUDA context進入不健康狀態(tài)。我在自研過程中踩過一次嚴重的泄漏某個實驗腳本在循環(huán)里反復創(chuàng)建和銷毀通信域但沒有正確關閉IPC handle跑了幾輪后顯存直線上升最后直接OOM。后來我給通信域增加了一個顯式的release接口統(tǒng)一關閉所有IPC handle并在進程退出信號處理里做了兜底清理。對于這種底層通信庫級別的代碼資源生命周期管理必須像對待文件句柄一樣認真不能覺得“進程退出系統(tǒng)會自動回收”就夠了。5. 實測對比兩種方案在不同消息尺寸下的真實差異5.1 盡量公平的測試方法為了對比NCCL Ring LL和自研PCIe IPC AllReduce我在同一臺8卡機器上跑了大量測試。硬件環(huán)境是近期常見的主流雙路服務器GPU通過PCIe switch互聯(lián)沒有NVLink橋接。測試方法很簡單固定一個消息尺寸讓兩種方案各自執(zhí)行100次AllReduce統(tǒng)計平均延遲和有效聚合帶寬然后遍歷從4KB到1GB的消息尺寸。需要特別說明的是這個對比不是“誰更厲害”的比拼而是要看不同場景下的適用性。NCCL是通用庫面對的是各種規(guī)模、各種拓撲的集群我的自研方案只針對這一臺機器做了定制。兩者不在同一個通用性層級但放在同一臺機器上比延遲和帶寬本身能說明很多問題。5.2 延遲與吞吐數(shù)據(jù)典型結果如下表所示數(shù)據(jù)是我當時記錄的一組有代表性的結果消息尺寸NCCL Ring LL平均延遲IPC AllReduce平均延遲IPC相對表現(xiàn)4KB約42微秒約31微秒延遲降低約26%64KB約85微秒約79微秒略優(yōu)1MB約420微秒約350微秒延遲降低約17%16MB約2.1毫秒約1.7毫秒延遲降低約19%256MB約33毫秒約28毫秒延遲降低約15%1GB約130毫秒約118毫秒延遲降低約9%有效聚合帶寬方面在256MB消息下NCCL實測約16.2GB/sIPC方案約18.5GB/s差距約14%。在1GB大消息下兩者接近NCCL約17.3GB/sIPC方案約17.9GB/s差距縮小到3%左右。5.3 分尺寸結果分析為什么各有勝負小消息場景下IPC方案的延遲優(yōu)勢最明顯。根本原因在于省掉了NCCL協(xié)議狀態(tài)機的額外開銷以及用更輕量的flag輪詢替代了部分同步機制。4KB消息的絕對時間雖然只有幾十微秒但在這個量級上協(xié)議開銷占比大減少的開銷直接變成了延遲收益。中消息場景16MB到256MB下IPC方案的優(yōu)勢依然存在。這個區(qū)間數(shù)據(jù)搬運已經占到主導但NCCL需要兼容NVLink、PCIe、網絡等多種路徑某些buffer策略和協(xié)議切換會引入額外的內存拷貝或等待。而我的方案完全針對PCIe拓撲定制數(shù)據(jù)塊大小和環(huán)序都做了適配所以能少繞一些路。但到了1GB這種超大消息兩種方案差距明顯縮小。原因也不難理解這時候協(xié)議開銷已經被海量數(shù)據(jù)傳輸攤薄真正比拼的是PCIe總線的物理帶寬上限和內存拷貝效率。NCCL的底層實現(xiàn)同樣經過高度優(yōu)化在純帶寬場景下自研方案的小聰明已經沒什么空間了。這個結果讓我確認一件事如果你主要跑小batch、小消息的同步訓練自研方案的收益非?？捎^如果你的模型大得每次梯度同步都是幾百MB甚至GB級別那直接用NCCL問題也不大自研方案的優(yōu)勢會被物理帶寬極限淹沒。6. 自研路上真正難住我的幾個坑6.1 對等映射撞上BAR1資源上限一個讓我印象極深的坑發(fā)生在嘗試映射超大通信buffer池時。理論上CUDA IPC允許你把一塊GPU顯存映射到另一個進程但底層需要通過PCIe BAR窗口來訪問對等內存。每張GPU的BAR1大小是有限的我遇到的情況是通信buffer池超過某個值之后cuIpcOpenMemHandle開始返回資源不足錯誤。排查過程很曲折。一開始以為是顯存不夠但檢查發(fā)現(xiàn)顯存明明很充足。后來才意識到是BAR1窗口大小的限制問題。我的應對方案是把通信buffer池拆成多個小塊分多次映射而不是一次性映射一整塊大緩沖區(qū)。雖然增加了管理復雜度但避免了BAR1的容量限制。這個坑對于任何走CUDA IPC做大數(shù)據(jù)量通信的人都可能遇到。如果你在映射時碰到資源不足相關錯誤先別急著懷疑顯存大小查一下BAR1窗口和當前已映射的對等內存總量。6.2 遺留的GPU上下文讓顯存悄悄“蒸發(fā)”前面提到過IPC handle泄漏問題但還有一個更隱蔽的變種GPU上下文沒有正確釋放。在CUDA里每個進程的GPU context都占用一定的顯存資源。如果我在調試時頻繁fork子進程、或者通過某種方式重復初始化CUDA而沒有主動銷毀context顯存就會一點點被吃光。我遇到過最夸張的一次是一個調試腳本跑了幾百輪后顯存從40GB可用掉到不足2GB但顯式分配的對象看起來都沒問題。最后發(fā)現(xiàn)是context泄漏堆積導致每個殘留context各自保留了底層通信buffer的引用。解決方式粗暴但有效給每個通信域加上顯式的context生命周期管理確保每個進程結束時CUDA資源能被徹底回收。這類問題在自研底層通信代碼時非常容易出現(xiàn)因為你不再依賴庫去管理通信資源所有上下文、映射、buffer都得自己盯著。6.3 對端進程崩潰后flag輪詢直接卡死另一個讓我花了很長時間的問題是“死等”。在IPC通信方案里接收方自旋等待發(fā)送方的flag變化這是性能優(yōu)勢的來源也是魯棒性的災難。一旦發(fā)送方進程在寫入flag之前崩潰接收方的自旋循環(huán)就永遠等不到目標值整個訓練進程直接hang住。我一開始的代碼沒有做任何超時處理結果在一次模擬故障測試中整個集群原地卡死只能手動kill。后來我在flag輪詢循環(huán)里增加了一個基于全局時鐘或固定迭代次數(shù)的超時判斷一旦超過閾值就報錯并觸發(fā)通信域重建。雖然看似增加了復雜度但在真實的生產環(huán)境里這套兜底機制是必須的。6.4 最后的保險保留NCCL回退開關自研方案上了之后我沒有把所有代碼的通信路徑都改成IPC實現(xiàn)。原因很簡單自研方案再優(yōu)化也只是針對我這臺機器的拓撲一旦換到別的機器所有定制都要重來。而NCCL作為通用庫幾乎在任何硬件組合上都能跑出一個可接受的結果。所以我在代碼里加了一個環(huán)境變量開關默認走IPC自研路徑但隨時可以切回NCCL。這個開關幫了大忙對比測試時用它保持公平在一些臨時搭建的測試機上直接用NCCL兜底調優(yōu)時也能快速確認問題到底出在通信層還是上層的模型代碼。最后說一點個人體會。做這個自研方案最值錢的不是省下來的幾十微秒延遲而是把一個黑盒通信庫拆開看的過程。NCCL的Ring LL已經足夠優(yōu)秀但只有當你自己動手推演過數(shù)據(jù)塊怎么切、flag怎么同步、PCIe帶寬怎么分配之后才能真的看懂它在背后替你處理了多少事。如果你也面臨單機多卡通信瓶頸我的建議是先別急著換機器花點時間測一下拓撲帶寬矩陣再決定是調NCCL參數(shù)還是自己動手。兩張卡之間的真實距離往往比PPT上的拓撲圖誠實得多。