DeepSeek/DeepEP
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepEP synced 2025-06-26 18:28:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #151 from vicoooo26/feat/nvidia-peer-mem-detection

Browse Source 
Feat: enhance nvidia peer memory detection
This commit is contained in:
Shangyan Zhou
2025-05-12 09:32:47 +08:00
committed by GitHub
parent f0a9f10629 c6051f3880
commit 05104029fd
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
third-party/nvshmem.patch vendored
View File

@@ -421,3 +421,17 @@ index ea1e284..e6640d6 100644
--
2.25.1
diff --git a/src/modules/transport/common/transport_ib_common.cpp b/src/modules/transport/common/transport_ib_common.cpp
index c89f408..f99018a 100644
--- a/src/modules/transport/common/transport_ib_common.cpp
+++ b/src/modules/transport/common/transport_ib_common.cpp
@@ -26,6 +26,9 @@ int nvshmemt_ib_common_nv_peer_mem_available() {
if (access("/sys/kernel/mm/memory_peers/nvidia-peermem/version", F_OK) == 0) {
return NVSHMEMX_SUCCESS;
}
+ if (access("/sys/module/nvidia_peermem/version", F_OK) == 0) {
+ return NVSHMEMX_SUCCESS;
+ }
return NVSHMEMX_ERROR_INTERNAL;
}