Fixes for NCCL_MAX_NCHANNELS and topo_expl (#398)

2021-06-22 08:41:49 -07:00
parent 720374a767
commit fa6d7e9a63
2 changed files with 2 additions and 1 deletions
@@ -292,6 +292,7 @@ ncclResult_t ncclTopoPostset(struct ncclComm* comm, int* firstRanks, int* treePa

  // Get number of channels after duplication
  nc *= comm->nChannels;
+  nc = std::min((int)ncclMaxNchannels(), nc);
  // Duplication should be complete now
  nChannels = comm->nChannels = std::min(MAXCHANNELS,nChannels*2);