From bb58b1c258a0bd1726ca6eed33f8e89552f7cf20 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Fri, 3 May 2024 16:42:07 -0700
Subject: [PATCH] Fix ignore NUMA not being observed for NICs during model
 matching (#1164)

---
 src/graph/rome_models.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/graph/rome_models.cc b/src/graph/rome_models.cc
index 387bbd69e2..278ea5cbfe 100644
--- a/src/graph/rome_models.cc
+++ b/src/graph/rome_models.cc
@@ -1393,7 +1393,7 @@ ncclResult_t parseRome4P2H(struct ncclTopoSystem* system, struct ncclTopoGraph*
             if (k == n[f]) break;
           if (f < found) continue;
           // check NUMA
-          if (romeTopoModels[i].nicNuma[j] != romeTopo.nicNuma[k]) continue;
+          if (!ignore_numa && romeTopoModels[i].nicNuma[j] != romeTopo.nicNuma[k]) continue;
           int g;
           // check GDR
           for (g = 0; g < ngpus; g++)