Increase P2P channels per peer (#1060)

[ROCm/rccl commit: ffde530af5]
Αυτή η υποβολή περιλαμβάνεται σε:
Wenkai Du authored and GitHub committed 2024-01-25 11:21:58 -08:00
1 γονέας b229abc692
υποβολή 71a356670f
1 file changed
+1
@@ -861,6 +861,7 @@ ncclResult_t ncclTopoComputeP2pChannels(struct ncclComm* comm) {
} else {
// Round to next pow2 nChannelsPerPeer and nChannels
comm->p2pnChannelsPerPeer = (ncclParamNChannelsPerPeer() == -2 ? nextPow2(minChannels) : ncclParamNChannelsPerPeer());
comm->p2pnChannelsPerPeer *= 2;
comm->p2pnChannels = nextPow2(comm->p2pnChannels);
}