[PR #422] [MERGED] Make dtype of topk_idx configurable #559

Closed
opened 2026-10-27 21:04:02 +00:00 by bsakaguchi · 0 comments
Owner

📋 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)

  • 6aa445c Make 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.

## 📋 Pull Request Information **Original PR:** https://github.com/deepseek-ai/DeepEP/pull/422 **Author:** [@yuantailing](https://github.com/yuantailing) **Created:** 9/23/2025 **Status:** ✅ Merged **Merged:** 9/24/2025 **Merged by:** [@sphish](https://github.com/sphish) **Base:** `main` ← **Head:** `topk_idx_t` --- ### 📝 Commits (1) - [`6aa445c`](https://github.com/deepseek-ai/DeepEP/commit/6aa445c3b8efac860a47e4893c69392a1f6bd423) Make dtype of topk_idx configurable ### 📊 Changes **14 files changed** (+78 additions, -50 deletions) <details> <summary>View changed files</summary> 📝 `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) </details> ### 📄 Description This is the 2nd part of #358 --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>
bsakaguchi 2026-10-27 21:04:02 +00:00
Sign in to join this conversation.
No labels
pull-request
No milestone
No project
No assignees
1 participant
Notifications
Due date
The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference
deepseek-ai/DeepEP#559
No description provided.