We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9b8a4e7 commit 64b197bCopy full SHA for 64b197b
torch/export/_remove_auto_functionalized_pass.py
@@ -1,4 +1,3 @@
1
-# mypy: allow-untyped-defs
2
# Copyright (c) Meta Platforms, Inc. and affiliates.
3
# All rights reserved.
4
#
@@ -15,7 +14,7 @@
15
14
from torch.export import ExportedProgram
16
17
18
-def remove_self_clone(graph: torch.fx.Graph):
+def remove_self_clone(graph: torch.fx.Graph) -> None:
19
for node in graph.nodes:
20
if node.target == torch.ops.aten.copy_.default and node.args[0] == node.args[1]:
21
node.replace_all_uses_with(node.args[0])
0 commit comments