[opt kimi k2 1 / n] Add kimi k2 moe fused gate (#13287)

2025-11-15 17:14:19 +08:00
parent 8e9f05ece1
commit 1d3d42bda0
8 changed files with 646 additions and 0 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -85,6 +85,7 @@ from sgl_kernel.moe import (
    apply_shuffle_mul_sum,
    cutlass_fp4_group_mm,
    fp8_blockwise_scaled_grouped_mm,
+    kimi_k2_moe_fused_gate,
    moe_align_block_size,
    moe_fused_gate,
    moe_sum,