Add various optimizations and Mega MoE benchmarks (#316)

* Merge with private repo * Add Mega MoE Benchmark * Minor fix * Update --------- Co-authored-by: Chenggang Zhao <chenggangz@deepseek.com>
2026-04-24 18:41:37 +08:00
parent 7f2a703ed5
commit 891d57b4db
21 changed files with 1276 additions and 372 deletions
--- a/deep_gemm/init.py
+++ b/deep_gemm/init.py
@@ -123,4 +123,4 @@ _C.init(
    _find_cuda_home()                           # CUDA home
 )

-__version__ = '2.4.2'
+__version__ = '2.5.0'