From 45f3fbc52f1df6ea0c48bc91270d7df77fca7b0e Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Wed, 3 Jul 2024 15:53:15 -0700
Subject: [PATCH] Checking kernel header files only when missing sysfs entry
 (#1239)

---
 src/transport/net_ib.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/transport/net_ib.cc b/src/transport/net_ib.cc
index 37bd6fab4d..5691f77c64 100644
--- a/src/transport/net_ib.cc
+++ b/src/transport/net_ib.cc
@@ -362,10 +362,9 @@ ncclResult_t ncclIbGdrSupport() {
       NCCLCHECK(ncclTopoGetStrFromSys("/proc/sys/kernel", "numa_balancing", strValue));
       if (strcmp(strValue, "1") == 0 && roMode == 0)
         moduleLoaded = 0;
-    } else {
+    } else if (moduleLoaded == 0) {
       char kernel_header_file[256];
       struct utsname utsname;
-      moduleLoaded = 0;
       char buf[256];
       FILE *fp = NULL;
       //check for kernel name exists