Support moe topk sigmoid kernel (#13049)

Co-authored-by: xuebi <xuebi@minimaxi.com>
2025-11-20 00:24:37 +08:00
parent 196b940aed
commit e72cf13693
10 changed files with 992 additions and 0 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -91,6 +91,7 @@ from sgl_kernel.moe import (
    moe_sum,
    moe_sum_reduce,
    prepare_moe_input,
+    topk_sigmoid,
    topk_softmax,
 )
 from sgl_kernel.quantization import (