Doubling P2P channels per peer on single node only (#1069)

[ROCm/rccl commit: 1d989f6524]
このコミットが含まれているのは:
Wenkai Du authored and GitHub committed 2024-02-02 12:41:00 -08:00
1 親 f23716de80
コミット 57e508f2e4
1ファイル変更
+2 -1
+2 -1
ファイルの表示
@@ -861,7 +861,8 @@ ncclResult_t ncclTopoComputeP2pChannels(struct ncclComm* comm) {
} else {
// Round to next pow2 nChannelsPerPeer and nChannels
comm->p2pnChannelsPerPeer = (ncclParamNChannelsPerPeer() == -2 ? nextPow2(minChannels) : ncclParamNChannelsPerPeer());
comm->p2pnChannelsPerPeer *= 2;
// Doubling P2P channels per peer on single node
if (comm->topo->nodes[GPU].count == comm->topo->nRanks) comm->p2pnChannelsPerPeer *= 2;
comm->p2pnChannels = nextPow2(comm->p2pnChannels);
}