[Bugfix] Fix benchmark_moe.py after inplace mechanism removal (#44041)

Signed-off-by: Qiuyang Yue <[email protected]>
This commit is contained in:
Qiuyang Yue
2026-06-07 00:32:00 -04:00
committed by GitHub
parent 6181e80fe0
commit 9c7f7741d4
-3
View File
@@ -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,
)