mirror of
https://github.com/deepseek-ai/DeepEP.git
synced 2026-11-05 00:47:32 +00:00
[PR #217] [MERGED] Support cuMem API in cross process shared memory management #471
Labels
No labels
pull-request
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
deepseek-ai/DeepEP#471
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
📋 Pull Request Information
Original PR: https://github.com/deepseek-ai/DeepEP/pull/217
Author: @fzyzcjy
Created: 6/17/2025
Status: ✅ Merged
Merged: 11/4/2025
Merged by: @LyricZhao
Base:
main← Head:feat/cu_mem_api📝 Commits (10+)
443bfa8moreb986ccemore3ea6f58more5d3513bmorebda5695more3740762moread4aee8moreb5e4aadmore240d058more5379d59more📊 Changes
7 files changed (+185 additions, -17 deletions)
View changed files
📝
csrc/deep_ep.cpp(+127 -12)📝
csrc/deep_ep.hpp(+33 -2)📝
csrc/kernels/exception.cuh(+12 -0)📝
deep_ep/buffer.py(+3 -1)📝
format.sh(+4 -0)📝
setup.py(+1 -1)📝
tests/test_intranode.py(+5 -1)📄 Description
code is roughly like this; I will work on related things more which will also verify this PR more
EDIT: it works well on the target hardware; I will try to find out some time to beautify and generalize the code (probably some time later)
🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.