GDR enablement logic fix for kernel 6.4.0+ (#1378)

[ROCm/rccl commit: 6178556853]
2024-11-02 23:20:07 -07:00
parent 251df02d42
commit 595cda2ab9
1 changed files with 3 additions and 1 deletions
@@ -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];