graph: fix multi-node minChannel count

2024-06-04 15:22:38 -05:00
parent 955347bab4
commit 6ab20a7c6b
1 changed files with 3 additions and 5 deletions
@@ -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())) {