diff --git a/mlir/lib/Conversion/SPIRVToLLVM/ConvertLaunchFuncToLLVMCalls.cpp b/mlir/lib/Conversion/SPIRVToLLVM/ConvertLaunchFuncToLLVMCalls.cpp index 035f197b1eac2..399ccf3925f3a 100644 --- a/mlir/lib/Conversion/SPIRVToLLVM/ConvertLaunchFuncToLLVMCalls.cpp +++ b/mlir/lib/Conversion/SPIRVToLLVM/ConvertLaunchFuncToLLVMCalls.cpp @@ -267,9 +267,8 @@ class GPULaunchLowering : public ConvertOpToLLVMPattern { copyInfo.push_back(info); } // Create a call to the kernel and copy the data back. - Operation *callOp = rewriter.replaceOpWithNewOp( - op, kernelFunc, ArrayRef()); - rewriter.setInsertionPointAfter(callOp); + rewriter.replaceOpWithNewOp(op, kernelFunc, + ArrayRef()); for (CopyInfo info : copyInfo) copy(loc, info.src, info.dst, info.size, rewriter); return success(); diff --git a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp index 6f288495516f7..0cb0badacf69d 100644 --- a/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp +++ b/mlir/lib/Conversion/TosaToLinalg/TosaToLinalgNamed.cpp @@ -802,7 +802,6 @@ class MaxPool2dConverter : public OpConversionPattern { ValueRange{paddedInput, fakeWindowDims}, filledEmptyTensor, strideAttr, dilationAttr); - rewriter.setInsertionPointAfter(op); NanPropagationMode nanMode = op.getNanMode(); rewriter.replaceOp(op, resultOp);