mirror of
https://github.com/deepseek-ai/DeepEP.git
synced 2026-11-05 00:47:32 +00:00
[PR #218] Support other NVLink scenarios #474
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#474
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/218
Author: @fzyzcjy
Created: 6/17/2025
Status: 🔄 Open
Base:
main← Head:feat/deepep_normal_update📝 Commits (10+)
be2eed8more9ecb941more9683d94more8cf6bd8moreacf108amore3e2cedeRevert "more"45fa1afRevert "more"443bfa8moreb986ccemore3ea6f58more📊 Changes
10 files changed (+248 additions, -21 deletions)
View changed files
📝
csrc/deep_ep.cpp(+135 -9)📝
csrc/deep_ep.hpp(+33 -1)📝
csrc/kernels/configs.cuh(+7 -0)📝
csrc/kernels/exception.cuh(+12 -0)📝
csrc/kernels/internode.cu(+6 -0)📝
csrc/kernels/launch.cuh(+2 -0)📝
csrc/kernels/layout.cu(+33 -2)📝
setup.py(+1 -1)📝
tests/test_intranode.py(+16 -7)📝
tests/utils.py(+3 -1)📄 Description
test with 4 rank, 8 rank, 24 rank all pass. code is messy and I will refactor later, and also will try to improve a bit of performance if possible.
Please subtract code diff from https://github.com/deepseek-ai/DeepEP/pull/217.
🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.