Force ring algorithm on single node (#509)

[ROCm/rccl commit: d6d6af710e]
Этот коммит содержится в:
Wenkai Du
2022-03-04 10:29:02 -08:00
коммит произвёл GitHub
родитель 211ff286be
Коммит 133aed2dfb
+1 -1
Просмотреть файл
@@ -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;