fix benchmark fp8 blockwise group gemm (#8815)

1bd53168 · Yuan Luo · GitHub · aeac900c · 1bd53168
Unverified Commit 1bd53168 authored Aug 06, 2025 by Yuan Luo Committed by GitHub Aug 06, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

sgl-kernel/benchmark/bench_fp8_blockwise_group_gemm.py sgl-kernel/benchmark/bench_fp8_blockwise_group_gemm.py +1 -3

No files found.
--- 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):