diff --git a/benchmarks/kernels/benchmark_moe.py b/benchmarks/kernels/benchmark_moe.py index 4463a23772..f7079af359 100644 --- a/benchmarks/kernels/benchmark_moe.py +++ b/benchmarks/kernels/benchmark_moe.py @@ -271,7 +271,6 @@ def benchmark_config( moe_config=moe_config, quant_config=quant_config, ), - inplace=not disable_inplace(), ) with override_config(config): @@ -279,7 +278,6 @@ def benchmark_config( x, input_gating, topk, renormalize=not use_deep_gemm ) - inplace = not disable_inplace() if use_deep_gemm: return deep_gemm_experts.apply( x, @@ -298,7 +296,6 @@ def benchmark_config( w2, topk_weights, topk_ids, - inplace=inplace, quant_config=quant_config, )