diff --git a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp index c460a8bb2f4b2..ca8f6fcb1e122 100644 --- a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp +++ b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp @@ -815,6 +815,7 @@ class MaxPool2dConverter : public OpConversionPattern { ValueRange{paddedInput, fakeWindowDims}, filledEmptyTensor, strideAttr, dilationAttr); + rewriter.setInsertionPointAfter(op); rewriter.replaceOp(op, resultOp); // NaN propagation has no meaning for non floating point types.