diff --git a/mlir/lib/Conversion/ShapeToStandard/ShapeToStandard.cpp b/mlir/lib/Conversion/ShapeToStandard/ShapeToStandard.cpp --- a/mlir/lib/Conversion/ShapeToStandard/ShapeToStandard.cpp +++ b/mlir/lib/Conversion/ShapeToStandard/ShapeToStandard.cpp @@ -55,8 +55,7 @@ matchAndRewrite(SrcOpTy op, ArrayRef operands, ConversionPatternRewriter &rewriter) const override { typename SrcOpTy::Adaptor adaptor(operands); - rewriter.replaceOpWithNewOp(op.getOperation(), adaptor.lhs(), - adaptor.rhs()); + rewriter.replaceOpWithNewOp(op, adaptor.lhs(), adaptor.rhs()); return success(); } }; @@ -123,8 +122,7 @@ LogicalResult ConstSizeOpConverter::matchAndRewrite( ConstSizeOp op, ArrayRef operands, ConversionPatternRewriter &rewriter) const { - rewriter.replaceOpWithNewOp(op.getOperation(), - op.value().getSExtValue()); + rewriter.replaceOpWithNewOp(op, op.value().getSExtValue()); return success(); } @@ -171,7 +169,7 @@ RankOpConverter::matchAndRewrite(shape::RankOp op, ArrayRef operands, ConversionPatternRewriter &rewriter) const { shape::RankOp::Adaptor transformed(operands); - rewriter.replaceOpWithNewOp(op.getOperation(), transformed.shape(), 0); + rewriter.replaceOpWithNewOp(op, transformed.shape(), 0); return success(); }