From 38f25e87fc9fe712c71308f87852cfc54a2481a7 Mon Sep 17 00:00:00 2001 From: penguin_wwy <940375606@qq.com> Date: Sat, 22 Mar 2025 15:52:34 +0800 Subject: [PATCH] Correcting default configuration when benchmarking fused_moe (#4665) --- benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py b/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py index 3a5dac7e2..89309ac6e 100644 --- a/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py +++ b/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py @@ -267,6 +267,7 @@ class BenchmarkWorker: topk, dtype_str, False, + block_shape, ) else: config = op_config[min(op_config.keys(), key=lambda x: abs(x - num_tokens))]