Correcting default configuration when benchmarking fused_moe (#4665)

38f25e87 · penguin_wwy · GitHub · 8cd42504 · 38f25e87
Unverified Commit 38f25e87 authored Mar 22, 2025 by penguin_wwy Committed by GitHub Mar 22, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py ...hmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py +1 -0

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