Fix positional argument (#7093)

2025-06-12 09:31:13 +08:00
parent 021f76e4f4
commit ef32677444
2 changed files with 2 additions and 0 deletions
--- a/benchmark/kernels/fused_moe_triton/benchmark_ep_pre_reorder_triton.py
+++ b/benchmark/kernels/fused_moe_triton/benchmark_ep_pre_reorder_triton.py
@@ -48,6 +48,7 @@ def benchmark_pre_reorder(batch_size, topk, model_config):
            topk,
            hidden_size,
            block_size,
+            use_per_token_if_dynamic=True,
        )

    for _ in range(10):