LL128 check if all XGMI (#1089)

[ROCm/rccl commit: b275ed0b56]
Tá an tiomantas seo le fáil i:
Bertan Dogancay údaraithe agus GitHub tiomanta 2024-02-21 09:41:40 -07:00
1 tuismitheoir bf48d1bc4d
tiomantas 4b4bdd904e
1 comhad athraithe
+1
+1
Féach ar an gComhad
@@ -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) {