From 4b4bdd904eae82cd0e2c3feefd9011e3c83fe63e Mon Sep 17 00:00:00 2001
From: Bertan Dogancay <111835151+BertanDogancay@users.noreply.github.com>
Date: Wed, 21 Feb 2024 09:41:40 -0700
Subject: [PATCH] LL128 check if all XGMI (#1089)

[ROCm/rccl commit: b275ed0b5653a42e9a9f45a908ba0ecc7b120659]
---
 projects/rccl/src/enqueue.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/projects/rccl/src/enqueue.cc b/projects/rccl/src/enqueue.cc
index f3f97d56bb..746da0e1c7 100644
--- a/projects/rccl/src/enqueue.cc
+++ b/projects/rccl/src/enqueue.cc
@@ -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) {