mirror of
https://github.com/deepseek-ai/DeepEP.git
synced 2026-11-05 00:47:32 +00:00
[PR #422] [MERGED] Make dtype of topk_idx configurable #559
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#559
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/422
Author: @yuantailing
Created: 9/23/2025
Status: ✅ Merged
Merged: 9/24/2025
Merged by: @sphish
Base:
main← Head:topk_idx_t📝 Commits (1)
6aa445cMake dtype of topk_idx configurable📊 Changes
14 files changed (+78 additions, -50 deletions)
View changed files
📝
csrc/config.hpp(+2 -2)📝
csrc/deep_ep.cpp(+15 -14)📝
csrc/kernels/api.cuh(+9 -7)📝
csrc/kernels/configs.cuh(+14 -0)📝
csrc/kernels/internode.cu(+5 -5)📝
csrc/kernels/internode_ll.cu(+4 -4)📝
csrc/kernels/intranode.cu(+6 -6)📝
csrc/kernels/layout.cu(+2 -2)📝
deep_ep/__init__.py(+1 -1)📝
deep_ep/buffer.py(+9 -9)📝
setup.py(+6 -0)📝
tests/test_internode.py(+2 -0)📝
tests/test_intranode.py(+2 -0)📝
tests/test_low_latency.py(+1 -0)📄 Description
This is the 2nd part of #358
🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.