From de9ebd8a8bcf480893e9cb96b38f3d29ab7f51a0 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Thu, 11 Sep 2025 08:44:10 -0700
Subject: [PATCH] Treat PIX and PXB as same GDR distance (#1894)

---
 src/graph/rome_models.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/graph/rome_models.cc b/src/graph/rome_models.cc
index e436e4fb85..0acc62573f 100644
--- a/src/graph/rome_models.cc
+++ b/src/graph/rome_models.cc
@@ -1791,6 +1791,10 @@ static bool permuteNetIds(int *n, int *g, int s, int last, struct rcclRomeModel*
       for (j = 0; j < ref->nGpus; j++) {
         // enabling PXN override paths over PHB and SYS
         if (topo->gdrLevel[n[i]*ref->nGpus+g[j]] == PATH_PXN) continue;
+        // treat PIX and PXB as same
+        if ((ref->gdrLevel[i*ref->nGpus+j] == PATH_PXB && topo->gdrLevel[n[i]*ref->nGpus+g[j]] == PATH_PIX) ||
+          (ref->gdrLevel[i*ref->nGpus+j] == PATH_PIX && topo->gdrLevel[n[i]*ref->nGpus+g[j]] == PATH_PXB))
+          continue;
         if (ref->gdrLevel[i*ref->nGpus+j] != topo->gdrLevel[n[i]*ref->nGpus+g[j]]) break;
       }
       if (j < ref->nGpus) break;