deepseek-ai/DeepGEMM
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepGEMM.git synced 2026-11-05 00:47:26 +00:00
Code Issues 162 Projects Packages Wiki Activity

[PR #382] [CLOSED] Add MXFP4 MegaMoE with FP8 combine #405

New issue
Closed
opened 2026-10-26 21:17:20 +00:00 by bsakaguchi · 0 comments
bsakaguchi commented 2026-10-26 21:17:20 +00:00
Owner
Copy link

📋 Pull Request Information

Original PR: https://github.com/deepseek-ai/DeepGEMM/pull/382
Author: @zyongye
Created: 7/16/2026
Status: ❌ Closed

Base: main ← Head: pr377-mxfp4-fp8combine-gb300-validated

📝 Commits (1)

  • edc5c10 Add MXFP4 MegaMoE with FP8 combine

📊 Changes

14 files changed (+1147 additions, -143 deletions)

View changed files

📝 csrc/apis/mega.hpp (+33 -14)
📝 csrc/jit_kernels/heuristics/mega_moe.hpp (+25 -13)
📝 csrc/jit_kernels/impls/sm100_fp8_fp4_mega_moe.hpp (+48 -14)
📝 deep_gemm/include/deep_gemm/comm/barrier.cuh (+7 -4)
📝 deep_gemm/include/deep_gemm/common/math.cuh (+22 -0)
📝 deep_gemm/include/deep_gemm/impls/sm100_fp8_fp4_mega_moe.cuh (+295 -70)
📝 deep_gemm/include/deep_gemm/layout/mega_moe.cuh (+23 -8)
📝 deep_gemm/include/deep_gemm/ptx/tcgen05.cuh (+24 -0)
📝 deep_gemm/mega/__init__.py (+28 -5)
➕ docs/mega_moe_pr377_rewrite.md (+128 -0)
➕ scripts/compare_mega_moe_results.py (+42 -0)
➕ tests/bench_mega_moe_ab.py (+255 -0)
📝 tests/test_mega_moe.py (+67 -15)
➕ tests/test_mega_moe_ref.py (+150 -0)

📄 Description

No description provided

🔄 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/DeepGEMM/pull/382 **Author:** [@zyongye](https://github.com/zyongye) **Created:** 7/16/2026 **Status:** ❌ Closed **Base:** `main` ← **Head:** `pr377-mxfp4-fp8combine-gb300-validated` --- ### 📝 Commits (1) - [`edc5c10`](https://github.com/deepseek-ai/DeepGEMM/commit/edc5c10c7eae4f2a27cd9aa916ceed914a6f65bb) Add MXFP4 MegaMoE with FP8 combine ### 📊 Changes **14 files changed** (+1147 additions, -143 deletions) <details> <summary>View changed files</summary> 📝 `csrc/apis/mega.hpp` (+33 -14) 📝 `csrc/jit_kernels/heuristics/mega_moe.hpp` (+25 -13) 📝 `csrc/jit_kernels/impls/sm100_fp8_fp4_mega_moe.hpp` (+48 -14) 📝 `deep_gemm/include/deep_gemm/comm/barrier.cuh` (+7 -4) 📝 `deep_gemm/include/deep_gemm/common/math.cuh` (+22 -0) 📝 `deep_gemm/include/deep_gemm/impls/sm100_fp8_fp4_mega_moe.cuh` (+295 -70) 📝 `deep_gemm/include/deep_gemm/layout/mega_moe.cuh` (+23 -8) 📝 `deep_gemm/include/deep_gemm/ptx/tcgen05.cuh` (+24 -0) 📝 `deep_gemm/mega/__init__.py` (+28 -5) ➕ `docs/mega_moe_pr377_rewrite.md` (+128 -0) ➕ `scripts/compare_mega_moe_results.py` (+42 -0) ➕ `tests/bench_mega_moe_ab.py` (+255 -0) 📝 `tests/test_mega_moe.py` (+67 -15) ➕ `tests/test_mega_moe_ref.py` (+150 -0) </details> ### 📄 Description _No description provided_ --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>
bsakaguchi 2026-10-26 21:17:20 +00:00
Sign in to join this conversation.
No Branch/Tag specified
Branches Tags
main
public-sync-20261014
0930-release
nv_dev
fix-mega-moe-fence
public-release-2609
public-release-2607
public-release-2606
mega-update
public-release-260416
ko3n1g/ci/build-wheels
nv_dev_f8e8fb5
nv_dev_950d31a
nv_dev_67fc648
nv_dev_c491439
nv_dev_4ff3f54
v2.1.1.post3
v2.1.1.post2
v2.1.1.post1
v2.1.1
v2.1.0
Labels
Clear labels   
pull-request
Mirrored from GitHub Pull Request

No labels
pull-request
Milestone
Clear milestone
No items
No milestone
Projects
Clear projects
No items
No project
Assignees
Clear assignees
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/DeepGEMM#405
No description provided.