File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -820,7 +820,9 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
820820 return PassResult (graph_module , overall_modified )
821821
822822
823- class CommonReorderPasses :
823+ # The following class consolidates functions to reoder ops (i.e., either hoist
824+ # or sink some ops in the graph).
825+ class CadenceReorderOpsInGraph :
824826 passes = [
825827 # Hoist/sink nodes closer to their SSA def/use
826828 HoistOpsCloserToDefPass ,
@@ -830,13 +832,6 @@ class CommonReorderPasses:
830832 # nodes closer to their def/use.
831833 AdvanceQuantizeOpAboveDefChainPass ,
832834 PostponeDequantizeOpBelowUseChainPass ,
833- ]
834-
835-
836- # The following class consolidates functions to reoder ops (i.e., either hoist
837- # or sink some ops in the graph).
838- class CadenceReorderOpsInGraph :
839- passes = CommonReorderPasses .passes + [
840835 # These passes work on branches instead of linear chains to advance
841836 # quantize op beyond their def.
842837 AdvanceQuantizeOpAboveDefInBranchPass ,
You can’t perform that action at this time.
0 commit comments