Skip to content

Commit 85902c5

Browse files
eigen-kfacebook-github-bot
authored andcommitted
Gather common remove passes in one list. (pytorch#14781)
Summary: Pull Request resolved: pytorch#14781 Differential Revision: D83793087
1 parent 4d681cb commit 85902c5

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

backends/cadence/aot/remove_ops.py

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
import logging
1111
from dataclasses import dataclass, field
12-
from typing import cast, List, Optional, Sequence, Set
12+
from typing import cast, List, Optional, Sequence, Set, Type
1313

1414
import torch
1515
import torch.fx
@@ -926,19 +926,23 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
926926
return super().call(graph_module)
927927

928928

929+
class CommonRemovePasses:
930+
passes: List[Type[ExportPass]] = [
931+
RemoveCloneOpPass,
932+
RemoveAliasCopyOpPass,
933+
RemoveNopExpandOpPass,
934+
RemoveNopSliceOrViewOpPass,
935+
RemoveNopSelectOpPass,
936+
RemoveToOpsPass,
937+
RemoveZeroSizedCatArgsPass,
938+
]
929939
class CadenceRemoveNops:
930-
passes = [
940+
passes: List[Type[ExportPass]] = CommonRemovePasses.passes + [
931941
SimplifySliceOpPass,
932942
RemoveCloneOpsTransformImported,
933-
RemoveToOpsPass,
934943
RemoveNopRequantizeOpPass,
935-
RemoveZeroSizedCatArgsPass,
936-
RemoveNopSliceOrViewOpPass,
937-
RemoveNopExpandOpPass,
938944
RemoveZeroSizedConstantPadNd,
939-
RemoveCloneOpPass,
940945
RemoveContiguousOpPass,
941-
RemoveAliasCopyOpPass,
942946
RemoveNopMulOpPass,
943947
RemoveNopAddOpPass,
944948
RemoveNopLinalgVectorNormOpPass,

0 commit comments

Comments
 (0)