Skip to content

Commit 471769f

Browse files
eigen-kfacebook-github-bot
authored andcommitted
Gather common remove passes in one list.
Differential Revision: D83793087
1 parent 4d681cb commit 471769f

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

backends/cadence/aot/remove_ops.py

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -926,19 +926,23 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
926926
return super().call(graph_module)
927927

928928

929-
class CadenceRemoveNops:
929+
class CommonRemovePasses:
930930
passes = [
931+
RemoveCloneOpPass,
932+
RemoveAliasCopyOpPass,
933+
RemoveNopExpandOpPass,
934+
RemoveNopSliceOrViewOpPass,
935+
RemoveNopSelectOpPass,
936+
RemoveToOpsPass,
937+
RemoveZeroSizedCatArgsPass,
938+
]
939+
class CadenceRemoveNops:
940+
passes = 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)