AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
[ROCm/rccl commit: 6178556853]
This commit is contained in:
Abhishek Kulkarni
2024-11-02 23:20:07 -07:00
committed by GitHub
parent 251df02d42
commit 595cda2ab9
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/transport/net_ib.cc
+3 -1
View File
@@ -607,7 +607,9 @@ ncclResult_t ncclIbGdrSupport() {
NCCLCHECK(ncclTopoGetStrFromSys("/proc/sys/kernel", "numa_balancing", strValue)); NCCLCHECK(ncclTopoGetStrFromSys("/proc/sys/kernel", "numa_balancing", strValue));
if (strcmp(strValue, "1") == 0 && roMode == 0) if (strcmp(strValue, "1") == 0 && roMode == 0)
moduleLoaded = 0; moduleLoaded = 0;
} else if (moduleLoaded == 0) { }
if (moduleLoaded == 0) {
// Check for `ib_register_peer_memory_client` symbol in `/proc/kallsyms` // Check for `ib_register_peer_memory_client` symbol in `/proc/kallsyms`
// if your system uses native OS ib_peer module // if your system uses native OS ib_peer module
char buf[256]; char buf[256];