Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
DeepEP
Commits
c6051f38
Commit
c6051f38
authored
May 09, 2025
by
Vico Chu
Browse files
Feat: enhance nvidia peer memory detection
parent
9056a6db
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
14 additions
and
0 deletions
+14
-0
third-party/nvshmem.patch
third-party/nvshmem.patch
+14
-0
No files found.
third-party/nvshmem.patch
View file @
c6051f38
...
...
@@ -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;
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment