diff --git a/lib/Dialect/Torch/Transforms/FuseQuantizedOps.cpp b/lib/Dialect/Torch/Transforms/FuseQuantizedOps.cpp index 7870ff63c..38bc4d275 100644 --- a/lib/Dialect/Torch/Transforms/FuseQuantizedOps.cpp +++ b/lib/Dialect/Torch/Transforms/FuseQuantizedOps.cpp @@ -39,7 +39,8 @@ template <> struct QuantInfo { bool isQCommutingOp(mlir::Operation *op) { // if adding a new commuting op here, be sure to add a // RemoveUnused pattern for that op to clean up afterwards - return llvm::isa(op); + return llvm::isa(op); } // The following conversion takes patterns of the form [op0 -> MPTQT -> dequant @@ -372,11 +373,12 @@ public: RemoveUnused, RemoveUnused, RemoveUnused, RemoveUnused, - RemoveUnused, - QuantizeOperandsPastCommutingOps, + RemoveUnused, RemoveUnused, + RemoveUnused, + QuantizeOperandsPastCommutingOps, QuantizeOperandsPastCommutingOps, QuantizeOperandsPastCommutingOps, - QuantizeOperandsPastCommutingOps, + QuantizeOperandsPastCommutingOps, QuantizeAccumulator, QuantizeAccumulator, QuantizeResultLikeOperand, QuantizeBias>( context);