[Bugfix] Fix benchmark_moe.py inplace assertion with torch >= 2.9 (#34149)

Signed-off-by: Matthias Gehre <matthias.gehre@amd.com>
2026-02-11 04:58:56 +01:00
parent c9a1923bb4
commit 7a048ee65f
1 changed files with 3 additions and 2 deletions
--- a/benchmarks/kernels/benchmark_moe.py
+++ b/benchmarks/kernels/benchmark_moe.py
@@ -226,9 +226,10 @@ def benchmark_config(
                x, input_gating, topk, renormalize=not use_deep_gemm
            )

+            inplace = not disable_inplace()
            if use_deep_gemm:
                return deep_gemm_experts(
-                    x, w1, w2, topk_weights, topk_ids, inplace=True
+                    x, w1, w2, topk_weights, topk_ids, inplace=inplace
                )
            return fused_experts(
                x,
@@ -236,7 +237,7 @@ def benchmark_config(
                w2,
                topk_weights,
                topk_ids,
-                inplace=True,
+                inplace=inplace,
                quant_config=quant_config,
            )