From 1bd5316873ee0ce327a5e92c0dc6bc799ff0d59c Mon Sep 17 00:00:00 2001 From: Yuan Luo Date: Wed, 6 Aug 2025 21:02:21 +0800 Subject: [PATCH] fix benchmark fp8 blockwise group gemm (#8815) --- sgl-kernel/benchmark/bench_fp8_blockwise_group_gemm.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/sgl-kernel/benchmark/bench_fp8_blockwise_group_gemm.py b/sgl-kernel/benchmark/bench_fp8_blockwise_group_gemm.py index 2a0a8e410..6aa131244 100644 --- 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):