[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>
This commit is contained in:
@@ -592,6 +592,7 @@ def pplx_moe(
|
||||
prepare_finalize,
|
||||
experts,
|
||||
shared_experts,
|
||||
inplace=False,
|
||||
)
|
||||
|
||||
# Note: for now use_compile will error out if the problem size is
|
||||
|
||||
Reference in New Issue
Block a user