diff --git a/mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h b/mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h --- a/mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h +++ b/mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h @@ -26,7 +26,7 @@ /// Bufferize `op` and its nested ops. Bufferization decisions are stored in /// `state`. -LogicalResult bufferizeOp(Operation *op, BufferizationState &state); +LogicalResult bufferizeOp(Operation *op, const BufferizationState &state); /// Run Comprehensive Bufferize on the given op: Analysis + Bufferization LogicalResult diff --git a/mlir/lib/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.cpp b/mlir/lib/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.cpp --- a/mlir/lib/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.cpp +++ b/mlir/lib/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.cpp @@ -558,7 +558,7 @@ /// Rewrite pattern that bufferizes bufferizable ops. struct BufferizationPattern : public OpInterfaceRewritePattern { - BufferizationPattern(MLIRContext *context, BufferizationState &state, + BufferizationPattern(MLIRContext *context, const BufferizationState &state, PatternBenefit benefit = 1) : OpInterfaceRewritePattern(context, benefit), state(state) {} @@ -646,9 +646,8 @@ return success(); } -LogicalResult -mlir::linalg::comprehensive_bufferize::bufferizeOp(Operation *op, - BufferizationState &state) { +LogicalResult mlir::linalg::comprehensive_bufferize::bufferizeOp( + Operation *op, const BufferizationState &state) { // Bufferize the op and its nested ops. OwningRewritePatternList patterns(op->getContext()); patterns.add(op->getContext(), state); @@ -665,7 +664,5 @@ return failure(); if (options->testAnalysisOnly) return success(); - if (failed(bufferizeOp(op, state))) - return failure(); - return success(); + return bufferizeOp(op, state); }