Skip to content

Commit c29ce90

Browse files
eigen-kfacebook-github-bot
authored andcommitted
Move passes which replace the quant/dequant to optimized cadence ops to the common section. (pytorch#14486)
Summary: As stated in the summary. Reviewed By: hsharma35 Differential Revision: D82999617
1 parent d0f486a commit c29ce90

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

backends/cadence/aot/replace_ops.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2251,6 +2251,9 @@ class CommonReplacePasses:
22512251
ReplaceRepeatWithCatPass,
22522252
ReplaceFullLikeWithFullPass,
22532253
ReplaceAtenConvolutionWithCadenceConvolutionPass,
2254+
ReplacePT2QuantWithCadenceQuantPass,
2255+
ReplacePT2DequantWithCadenceDequantPass,
2256+
ReplacePowWithMulPass,
22542257
]
22552258

22562259

@@ -2296,13 +2299,10 @@ class CadenceReplaceOpsInGraph:
22962299
ReplaceScalarTensorWithFullPass,
22972300
ReplaceInfArgInFullWithValuePass,
22982301
ReplaceLogicalNotBooleanWhereWithWherePass,
2299-
ReplacePT2QuantWithCadenceQuantPass,
2300-
ReplacePT2DequantWithCadenceDequantPass,
23012302
ReplaceSingleElementTensorArgumentsFromFullOpWithScalarPass,
23022303
ReplaceAdaptiveAvgPoolWithAtenAvgPoolPass,
23032304
ReplaceAtenAvgPoolWithCadenceAvgPoolPass,
23042305
ReplaceWhereWithFullArgsWithWhereScalar,
23052306
ReplaceAtenApproxGeluWithApproxGeluPass,
2306-
ReplacePowWithMulPass,
23072307
ReplaceMulTensorWithMulAndFullOpsPass,
23082308
]

0 commit comments

Comments
 (0)