[PR #433] [MERGED] Add Jit compiler #566

New issue

Closed

opened 2026-10-27 21:04:07 +00:00 by bsakaguchi · 0 comments

bsakaguchi commented

2026-10-27 21:04:07 +00:00

Owner

📋 Pull Request Information

Original PR: https://github.com/deepseek-ai/DeepEP/pull/433
Author: @Autumn1998
Created: 9/26/2025
Status: ✅ Merged
Merged: 9/26/2025
Merged by: @jershi425

Base: hybrid-ep ← Head: mcore_dev

📝 Commits (5)

84ee983 reformat the code , prepare for the jit
80840a0 refactor and use remote allocater
27d96bb Add jit compiler, rm constexpr logic
8de95af license
efed1b1 rm empty file

📊 Changes

15 files changed (+1350 additions, -798 deletions)

View changed files

➖ csrc/hybrid_ep.cuh (+0 -270)
➕ csrc/hybrid_ep/allocator/allocator.cu (+122 -0)
➕ csrc/hybrid_ep/allocator/allocator.cuh (+47 -0)
📝 csrc/hybrid_ep/backend/hybrid_ep_backend.cuh (+13 -56)
➕ csrc/hybrid_ep/config.cuh (+60 -0)
📝 csrc/hybrid_ep/hybrid_ep.cu (+157 -261)
➕ csrc/hybrid_ep/hybrid_ep.cuh (+115 -0)
➕ csrc/hybrid_ep/jit/compiler.cu (+311 -0)
➕ csrc/hybrid_ep/jit/compiler.cuh (+98 -0)
➕ csrc/hybrid_ep/pybind_hybrid_ep.cu (+98 -0)
➕ csrc/hybrid_ep/utils.cuh (+127 -0)
➖ csrc/kernels/hybrid_ep_backend_configs.hpp (+0 -80)
📝 deep_ep/hybrid_ep_buffer.py (+122 -50)
📝 setup.py (+48 -25)
📝 tests/test_hybrid_ep.py (+32 -56)

📄 Description

Minor refactor
Add jit compiler
Remove the constexper logic

_{🔄 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/433 **Author:** [@Autumn1998](https://github.com/Autumn1998) **Created:** 9/26/2025 **Status:** ✅ Merged **Merged:** 9/26/2025 **Merged by:** [@jershi425](https://github.com/jershi425) **Base:** `hybrid-ep` ← **Head:** `mcore_dev` --- ### 📝 Commits (5) - [`84ee983`](https://github.com/deepseek-ai/DeepEP/commit/84ee983ebab632c78528b8e6e8ebca41b0412823) reformat the code , prepare for the jit - [`80840a0`](https://github.com/deepseek-ai/DeepEP/commit/80840a08d1117f65f59c0f537b4233c3b71e4241) refactor and use remote allocater - [`27d96bb`](https://github.com/deepseek-ai/DeepEP/commit/27d96bb3e23cabc8a44fac72f53fb687bc0f43b3) Add jit compiler, rm constexpr logic - [`8de95af`](https://github.com/deepseek-ai/DeepEP/commit/8de95af28f7a3af0b3f82e773cb12e380490ddee) license - [`efed1b1`](https://github.com/deepseek-ai/DeepEP/commit/efed1b1f4aeb115aba3ba92437937fc350a26e8e) rm empty file ### 📊 Changes **15 files changed** (+1350 additions, -798 deletions) <details> <summary>View changed files</summary> ➖ `csrc/hybrid_ep.cuh` (+0 -270) ➕ `csrc/hybrid_ep/allocator/allocator.cu` (+122 -0) ➕ `csrc/hybrid_ep/allocator/allocator.cuh` (+47 -0) 📝 `csrc/hybrid_ep/backend/hybrid_ep_backend.cuh` (+13 -56) ➕ `csrc/hybrid_ep/config.cuh` (+60 -0) 📝 `csrc/hybrid_ep/hybrid_ep.cu` (+157 -261) ➕ `csrc/hybrid_ep/hybrid_ep.cuh` (+115 -0) ➕ `csrc/hybrid_ep/jit/compiler.cu` (+311 -0) ➕ `csrc/hybrid_ep/jit/compiler.cuh` (+98 -0) ➕ `csrc/hybrid_ep/pybind_hybrid_ep.cu` (+98 -0) ➕ `csrc/hybrid_ep/utils.cuh` (+127 -0) ➖ `csrc/kernels/hybrid_ep_backend_configs.hpp` (+0 -80) 📝 `deep_ep/hybrid_ep_buffer.py` (+122 -50) 📝 `setup.py` (+48 -25) 📝 `tests/test_hybrid_ep.py` (+32 -56) </details> ### 📄 Description 1. Minor refactor 2. Add jit compiler 3. Remove the constexper logic --- <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:07 +00:00

closed this issue
added the
pull-request
label

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#566

No description provided.

Rows
Columns