diff --git a/backends/cadence/aot/reorder_ops.py b/backends/cadence/aot/reorder_ops.py index b1daefb8a2f..8a0e112aaf3 100644 --- a/backends/cadence/aot/reorder_ops.py +++ b/backends/cadence/aot/reorder_ops.py @@ -820,7 +820,9 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult: return PassResult(graph_module, overall_modified) -class CommonReorderPasses: +# The following class consolidates functions to reoder ops (i.e., either hoist +# or sink some ops in the graph). +class CadenceReorderOpsInGraph: passes = [ # Hoist/sink nodes closer to their SSA def/use HoistOpsCloserToDefPass, @@ -830,13 +832,6 @@ class CommonReorderPasses: # nodes closer to their def/use. AdvanceQuantizeOpAboveDefChainPass, PostponeDequantizeOpBelowUseChainPass, - ] - - -# The following class consolidates functions to reoder ops (i.e., either hoist -# or sink some ops in the graph). -class CadenceReorderOpsInGraph: - passes = CommonReorderPasses.passes + [ # These passes work on branches instead of linear chains to advance # quantize op beyond their def. AdvanceQuantizeOpAboveDefInBranchPass,