graph: fix minNchannels for multi-node

Multi-node rccl was not correctly setting the minNchannels value. This
PR fixes the bug.
Этот коммит содержится в:
Nusrat Islam
2024-06-24 16:42:38 -05:00
родитель 1ddb02c010
Коммит 05df0f8cea
+1 -1
Просмотреть файл
@@ -674,7 +674,7 @@ ncclResult_t ncclTopoPostset(struct ncclComm* comm, int* firstRanks, int* treePa
int minNchannels = ncclMinNchannels();
if (comm->nNodes > 1) {
minNchannels = std::min(64, maxChannels);
minNchannels = std::min(64, minNchannels);
}
if (mscclEnabled() && (comm->topo->mscclEnabled || mscclForceEnabled())) {