Only set minNchannels if we are actually using MSCCL, checked using comm->mscclCompatible. (#1337)

[ROCm/rccl commit: c11f6b1531]
Этот коммит содержится в:
corey-derochie-amd
2024-10-08 10:20:55 -06:00
коммит произвёл GitHub
родитель 949fdd027b
Коммит 35d98330f2
+3 -1
Просмотреть файл
@@ -691,7 +691,9 @@ ncclResult_t ncclTopoPostset(struct ncclComm* comm, int* firstRanks, int* treePa
if (mscclEnabled() && (comm->topo->mscclEnabled || mscclForceEnabled())) {
int mscclNumChannelsRequired = maxNchannels;
mscclSchedulerInit(comm, &mscclNumChannelsRequired);
minNchannels = std::max(minNchannels, mscclNumChannelsRequired);
if (comm->mscclCompatible) {
minNchannels = std::max(minNchannels, mscclNumChannelsRequired);
}
}
// Honor NCCL_MIN_NRINGS/NCCL_MAX_NRINGS.