diff --git a/optimum/exporters/executorch/recipes/xnnpack.py b/optimum/exporters/executorch/recipes/xnnpack.py index ef37cc7f..e224142d 100644 --- a/optimum/exporters/executorch/recipes/xnnpack.py +++ b/optimum/exporters/executorch/recipes/xnnpack.py @@ -75,6 +75,7 @@ def _lower_to_executorch( backend_config_dict = { "extract_delegate_segments": True, "memory_planning_pass": MemoryPlanningPass(alloc_graph_input=False), + "run_reinplace_pass": True, } backend_config_dict["do_quant_fusion_and_const_prop"] = True logging.debug(f"\nExported program: {exported_programs}")