Skip to content

Commit 64b197b

Browse files
bobrenjc93pytorchmergebot
authored andcommitted
remove allow-untyped-defs from export/_remove_auto_functionalized_pass.py (pytorch#144135)
Pull Request resolved: pytorch#144135 Approved by: https://github.com/Skylion007
1 parent 9b8a4e7 commit 64b197b

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

torch/export/_remove_auto_functionalized_pass.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
# mypy: allow-untyped-defs
21
# Copyright (c) Meta Platforms, Inc. and affiliates.
32
# All rights reserved.
43
#
@@ -15,7 +14,7 @@
1514
from torch.export import ExportedProgram
1615

1716

18-
def remove_self_clone(graph: torch.fx.Graph):
17+
def remove_self_clone(graph: torch.fx.Graph) -> None:
1918
for node in graph.nodes:
2019
if node.target == torch.ops.aten.copy_.default and node.args[0] == node.args[1]:
2120
node.replace_all_uses_with(node.args[0])

0 commit comments

Comments
 (0)