fix benchmark fp8 blockwise group gemm (#8815)

2025-08-06 21:02:21 +08:00
parent aeac900ca2
commit 1bd5316873
1 changed files with 1 additions and 3 deletions
--- a/sgl-kernel/benchmark/bench_fp8_blockwise_group_gemm.py
+++ b/sgl-kernel/benchmark/bench_fp8_blockwise_group_gemm.py
@@ -97,9 +97,7 @@ def bench_deepgemm(
    )

    def run_deepgemm():
-        deep_gemm.m_grouped_gemm_fp8_fp8_bf16_nt_contiguous(
-            x_fp8, y_fp8, out, m_indices
-        )
+        deep_gemm.m_grouped_fp8_gemm_nt_contiguous(x_fp8, y_fp8, out, m_indices)

    # warmup
    for _ in range(num_warmup):