From f679db6ff68e28414711c52e54539f4deda4bae9 Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Wed, 8 May 2024 06:59:57 -0700
Subject: [PATCH] Use normal permute path when one NIC per GPU (#1171)

---
 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 278ea5cbfe..347e8873b1 100644
--- a/src/graph/rome_models.cc
+++ b/src/graph/rome_models.cc
@@ -1379,7 +1379,7 @@ ncclResult_t parseRome4P2H(struct ncclTopoSystem* system, struct ncclTopoGraph*
     // permute GPU IDs
     for (int j = 0; j < ngpus; j++) g[j] = (j+2)%ngpus;
     if (!permuteGpuIds(g, 0, ngpus-1, romeTopoModels+i, &romeTopo, &time, ignore_cpu ? false : match_nbio, ignore_numa)) continue;
-    if (nnetspergpu) {
+    if (nnetspergpu > 1) {
       int found = 0;
       // initialize nics mapping
       for (int j = 0; j < nnets; j++) n[j] = -1;