Add copy_ to aten op lib. (#2202)

JacobSzwejbka · facebook-github-bot · commit 3609c7231b3e · 2024-03-04T12:30:44.000-08:00
Summary: Pull Request resolved: #2202 title bypass-github-export-checks neither failure in oss looks related to this diff so skipping Reviewed By: lucylq Differential Revision: D54432221 fbshipit-source-id: 0bff99ecaa6c7c2e403166e3b338cef157830b0d
diff --git a/exir/passes/__init__.py b/exir/passes/__init__.py
@@ -249,6 +249,7 @@ def callWithLoggerEnabled(self, graph_module: torch.fx.GraphModule) -> None:
     # it's retraced after running to_out_variant with the first trace.
     memory.alloc,
     executorch_call_delegate,
+    torch.ops.aten.copy_.default,
 }
 to_out_var_skiplist.update(_EXECUTORCH_SYM_OPS)
 
diff --git a/kernels/aten/functions.yaml b/kernels/aten/functions.yaml
@@ -15,6 +15,8 @@
 
 - op: _softmax.out
 
+- op: copy_
+
 - op: _to_copy.out
 
 - op: _unique2.out

Original file line number	Diff line number	Diff line change
`@@ -249,6 +249,7 @@ def callWithLoggerEnabled(self, graph_module: torch.fx.GraphModule) -> None:`
`249`	`249`	`# it's retraced after running to_out_variant with the first trace.`
`250`	`250`	`memory.alloc,`
`251`	`251`	`executorch_call_delegate,`
	`252`	`+ torch.ops.aten.copy_.default,`
`252`	`253`	`}`
`253`	`254`	`to_out_var_skiplist.update(_EXECUTORCH_SYM_OPS)`
`254`	`255`