[Moe Refactor] Make Inplace Flag for FusedMoEModularKernel part of the constructor (#33375)

Signed-off-by: Bill Nell <bnell@redhat.com> Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
2026-02-05 13:07:18 -05:00
parent 1ee95841bd
commit a57c8228ff
37 changed files with 132 additions and 109 deletions
--- a/tests/kernels/moe/test_pplx_cutlass_moe.py
+++ b/tests/kernels/moe/test_pplx_cutlass_moe.py
@@ -172,6 +172,7 @@ def pplx_cutlass_moe(
    fused_cutlass_experts = FusedMoEModularKernel(
        prepare_finalize,
        experts,
+        inplace=False,
    )

    a_chunk = chunk_by_rank(a, rank, world_size).to(device)