Skip to content

Commit b3de67a

Browse files
committed
address comments
1 parent d2c7c55 commit b3de67a

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

backends/aoti/passes/replace_view_copy_with_view.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -18,13 +18,6 @@
1818
from torch import fx
1919

2020

21-
_VIEW_COPY_TARGETS: Tuple[torch._ops.OpOverload | EdgeOpOverload] = (
22-
torch.ops.aten.slice_copy.Tensor,
23-
ops.edge.aten.slice_copy.Tensor,
24-
torch.ops.aten.select_copy.int,
25-
ops.edge.aten.select_copy.int,
26-
)
27-
2821
_VIEW_TARGETS: Dict[
2922
torch._ops.OpOverload | EdgeOpOverload, torch._ops.OpOverload | EdgeOpOverload
3023
] = {
@@ -42,7 +35,7 @@ def call(self, graph_module: fx.GraphModule) -> PassResult:
4235
graph_changed = False
4336

4437
for node in graph_module.graph.nodes:
45-
if node.op != "call_function" or node.target not in _VIEW_COPY_TARGETS:
38+
if node.op != "call_function" or node.target not in _VIEW_TARGETS:
4639
continue
4740

4841
if self._has_blocking_user(node, node.users.keys()):

0 commit comments

Comments
 (0)