LL128 check if all XGMI (#1089)

2024-02-21 09:41:40 -07:00
parent 5a0f9990a9
commit b275ed0b56
1 changed files with 1 additions and 0 deletions
@@ -1242,6 +1242,7 @@ static ncclResult_t topoGetAlgoInfo(struct ncclInfo* info, int collNetSupport, i
      if (a == NCCL_ALGO_NVLS_TREE && nvlsSupport != 1) continue;

      for (int p=0; p<NCCL_NUM_PROTOCOLS; p++) {
+        if (p == NCCL_PROTO_LL128 && info->comm->topo->type != RCCL_TOPO_XGMI_ALL) continue;
        float time;
        NCCLCHECK(ncclTopoGetAlgoTime(info, a, p, numPipeOps, &time, &backup));
        if (!backup) {