graph: fix multi-node minChannel count

[ROCm/rccl commit: 6ab20a7c6b]
Αυτή η υποβολή περιλαμβάνεται σε:
Nusrat Islam
2024-06-04 15:22:38 -05:00
γονέας 99fca5fa3a
υποβολή bf35178250
@@ -668,11 +668,9 @@ ncclResult_t ncclTopoPostset(struct ncclComm* comm, int* firstRanks, int* treePa
nChannels = comm->nChannels = copyChannels(comm, nChannels, 2*nChannels, ringPrev, ringNext);
}
int minNchannels = 64;
if (comm->nNodes == 1) {
minNchannels = ncclMinNchannels();
} else {
minNchannels = std::min(64,ncclMinNchannels());
int minNchannels = ncclMinNchannels();
if (comm->nNodes > 1) {
minNchannels = std::min(64, maxChannels);
}
if (mscclEnabled() && (comm->topo->mscclEnabled || mscclForceEnabled())) {