[Bugfix] Fix benchmark_moe.py (#23177)

jeejeelee · web-flow · commit 4d9c61993ac4 · 2025-08-19T13:39:40.000Z
Signed-off-by: Jee Jee Li &lt;pandaleefree@gmail.com&gt;
diff --git a/benchmarks/kernels/benchmark_moe.py b/benchmarks/kernels/benchmark_moe.py
@@ -430,7 +430,6 @@ def benchmark(
                 hidden_size,
                 topk,
                 dtype_str,
-                is_marlin=False,
             )
         else:
             config = op_config[min(op_config.keys(), key=lambda x: abs(x - num_tokens))]

Original file line number	Diff line number	Diff line change
`@@ -430,7 +430,6 @@ def benchmark(`
`430`	`430`	`hidden_size,`
`431`	`431`	`topk,`
`432`	`432`	`dtype_str,`
`433`		`- is_marlin=False,`
`434`	`433`	`)`
`435`	`434`	`else:`
`436`	`435`	`config = op_config[min(op_config.keys(), key=lambda x: abs(x - num_tokens))]`