Add dsv3 router gemm kernel (#7627)

2025-06-29 23:31:55 -07:00
parent 22352d47a9
commit 7248272ccc
8 changed files with 398 additions and 0 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -34,6 +34,7 @@ from sgl_kernel.gemm import (
    bmm_fp8,
    cutlass_scaled_fp4_mm,
    dsv3_fused_a_gemm,
+    dsv3_router_gemm,
    fp8_blockwise_scaled_mm,
    fp8_scaled_mm,
    int8_scaled_mm,