LL128 check if all XGMI (#1089)

[ROCm/rccl commit: b275ed0b56]
This commit is contained in:
Bertan Dogancay
2024-02-21 09:41:40 -07:00
committato da GitHub
parent bf48d1bc4d
commit 4b4bdd904e
+1
Vedi File
@@ -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) {