diff --git a/benchmarks/kernels/benchmark_moe.py b/benchmarks/kernels/benchmark_moe.py index b4a03665ef10..752c2d008216 100644 --- a/benchmarks/kernels/benchmark_moe.py +++ b/benchmarks/kernels/benchmark_moe.py @@ -430,7 +430,6 @@ class BenchmarkWorker: hidden_size, topk, dtype_str, - is_marlin=False, ) else: config = op_config[min(op_config.keys(), key=lambda x: abs(x - num_tokens))]