Merge remote-tracking branch 'rccl/develop' into 2.19.4

This commit is contained in:
Wenkai Du
2024-01-31 11:08:54 -06:00
14 changed files with 31 additions and 30 deletions
+1
View File
@@ -863,6 +863,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);
}