Force ring algorithm on single node (#509)

[ROCm/rccl commit: d6d6af710e]
This commit is contained in:
Wenkai Du authored and GitHub committed 2022-03-04 10:29:02 -08:00
1 förälder 211ff286be
incheckning 133aed2dfb
1 file changed
+1 -1
+1 -1
Visa fil
@@ -126,7 +126,7 @@ ncclResult_t ncclTopoTuneModel(struct ncclComm* comm, int minCompCap, int maxCom
nNodes;
for (int a=0; a<NCCL_NUM_ALGORITHMS; a++) {
if (coll != ncclFuncAllReduce && a != NCCL_ALGO_RING) continue;
if ((coll != ncclFuncAllReduce || nNodes == 1) && a != NCCL_ALGO_RING) continue;
for (int p=0; p<NCCL_NUM_PROTOCOLS; p++) {
float speed = nNodes <= 2 || a == NCCL_ALGO_COLLNET ? graphs[a]->speedIntra : graphs[a]->speedInter;