Unit Tests for testing channels (#1222)

Este commit está contenido en:
saurabhAMD
2024-06-25 10:10:10 -05:00
cometido por GitHub
padre 5f2b88bc28
commit e170f41ddd
Se han modificado 5 ficheros con 53 adiciones y 38 borrados
+5 -1
Ver fichero
@@ -674,7 +674,11 @@ ncclResult_t ncclTopoPostset(struct ncclComm* comm, int* firstRanks, int* treePa
int minNchannels = ncclMinNchannels();
if (comm->nNodes > 1) {
minNchannels = std::min(64, minNchannels);
minNchannels = std::min(64, maxChannels);
}
if (comm->nRanks < 8 && 64 < minNchannels) {
minNchannels = 2;
WARN("NCCL_MIN_NCHANNELS set by environment is ignored due to less than 8 GPUs.");
}
if (mscclEnabled() && (comm->topo->mscclEnabled || mscclForceEnabled())) {