Add dsv3 fused a gemm to sgl-kernel (#7630)

2025-06-29 17:52:24 +08:00
parent 071a1f51ae
commit 04b35190e2
9 changed files with 800 additions and 0 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -33,6 +33,7 @@ from sgl_kernel.gemm import (
    awq_dequantize,
    bmm_fp8,
    cutlass_scaled_fp4_mm,
+    dsv3_fused_a_gemm,
    fp8_blockwise_scaled_mm,
    fp8_scaled_mm,
    int8_scaled_mm,