From 638f10c669fcb1220bf1031ae6f480b5a87b3d63 Mon Sep 17 00:00:00 2001
From: Abhishek Kulkarni <11399+adk9@users.noreply.github.com>
Date: Mon, 14 Oct 2024 23:32:03 +0000
Subject: [PATCH] GDR enablement logic fix for kernel 6.4.0+

---
 src/transport/net_ib.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/transport/net_ib.cc b/src/transport/net_ib.cc
index e75106cf4..c5c2d6c49 100644
--- a/src/transport/net_ib.cc
+++ b/src/transport/net_ib.cc
@@ -607,7 +607,9 @@ ncclResult_t ncclIbGdrSupport() {
       NCCLCHECK(ncclTopoGetStrFromSys("/proc/sys/kernel", "numa_balancing", strValue));
       if (strcmp(strValue, "1") == 0 && roMode == 0)
         moduleLoaded = 0;
-    } else if (moduleLoaded == 0) {
+    }
+
+    if (moduleLoaded == 0) {
       // Check for `ib_register_peer_memory_client` symbol in `/proc/kallsyms`
       // if your system uses native OS ib_peer module
       char buf[256];