Skip to content

Commit 1402f6c

Browse files
digantdesaifacebook-github-bot
authored andcommitted
add remove clone in pre edge
Differential Revision: D83864591
1 parent 7116e0a commit 1402f6c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exir/passes/remove_noop_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ def call(self, graph_module: GraphModule) -> PassResult:
104104
if node.op != "call_function":
105105
continue
106106

107-
if node.target not in (torch.ops.aten._to_copy.default,):
107+
if node.target not in (torch.ops.aten._to_copy.default, torch.ops.aten.clone.default):
108108
continue
109109

110110
orig_tensor = node.args[0].meta["val"]

0 commit comments

Comments
 (0)