Deepcopy LoweredBackendModule meta (#1913)

angelayi · facebook-github-bot · commit 338fc6236359 · 2024-02-09T14:01:09.000-08:00
Summary: Pull Request resolved: #1913 Fixes T178315739 Reviewed By: mergennachin Differential Revision: D53615124 fbshipit-source-id: 2d5eaf134b418187f2b783ac174d96f9d650892f
diff --git a/exir/delegate.py b/exir/delegate.py
@@ -49,7 +49,7 @@ def _unwrap_proxy(e):
         )
 
     with disable_proxy_modes_tracing():
-        out = executorch_call_delegate(lowered_module, *args)
+        out = call_delegate_cpu(lowered_module, *args)
 
     get_lowered_module_name(proxy_mode.tracer.root, lowered_module)
 
diff --git a/exir/lowered_backend_module.py b/exir/lowered_backend_module.py
@@ -87,12 +87,14 @@ def __deepcopy__(self, memo: Optional[Dict[int, Any]]) -> "LoweredBackendModule"
             constants=self._original_module.constants,
         )
 
-        return LoweredBackendModule(
+        res = LoweredBackendModule(
             edge_program=copied_program,
             backend_id=self._backend_id,
             processed_bytes=self._processed_bytes,
             compile_specs=copy.deepcopy(self._compile_specs, memo),
         )
+        res.meta = copy.copy(getattr(self, "meta", {}))
+        return res
 
     @property
     def backend_id(self) -> str:

Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@ def _unwrap_proxy(e):`
`49`	`49`	`)`
`50`	`50`
`51`	`51`	`with disable_proxy_modes_tracing():`
`52`		`- out = executorch_call_delegate(lowered_module, *args)`
	`52`	`+ out = call_delegate_cpu(lowered_module, *args)`
`53`	`53`
`54`	`54`	`get_lowered_module_name(proxy_mode.tracer.root, lowered_module)`
`55`	`55`
Original file line number	Diff line number	Diff line change
`@@ -87,12 +87,14 @@ def __deepcopy__(self, memo: Optional[Dict[int, Any]]) -> "LoweredBackendModule"`
`87`	`87`	`constants=self._original_module.constants,`
`88`	`88`	`)`
`89`	`89`
`90`		`- return LoweredBackendModule(`
	`90`	`+ res = LoweredBackendModule(`
`91`	`91`	`edge_program=copied_program,`
`92`	`92`	`backend_id=self._backend_id,`
`93`	`93`	`processed_bytes=self._processed_bytes,`
`94`	`94`	`compile_specs=copy.deepcopy(self._compile_specs, memo),`
`95`	`95`	`)`
	`96`	`+ res.meta = copy.copy(getattr(self, "meta", {}))`
	`97`	`+ return res`
`96`	`98`
`97`	`99`	`@property`
`98`	`100`	`def backend_id(self) -> str:`