Fixes for NCCL_MAX_NCHANNELS and topo_expl (#398)

Tá an tiomantas seo le fáil i:
Wenkai Du
2021-06-22 08:41:49 -07:00
tiomanta ag GitHub
tuismitheoir 720374a767
tiomantas fa6d7e9a63
D'athraigh 2 comhad le 2 breiseanna agus 1 scriosta
+1
Féach ar an gComhad
@@ -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);