maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir][linalg][bufferize][NFC] Pass missing BufferizationState objs as const ref 

These should have been updated as part of D116742.

Differential Revision: https://reviews.llvm.org/D116937
This commit is contained in:
Matthias Springer 2022-01-10 23:36:46 +09:00
parent 1a2474b786
commit 0696ad9563
2 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlir/include/mlir/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.h
View File

@ -26,7 +26,7 @@ LogicalResult analyzeOp(Operation *op, BufferizationState &state);
/// 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

11
mlir/lib/Dialect/Linalg/ComprehensiveBufferize/ComprehensiveBufferize.cpp
View File

@ -558,7 +558,7 @@ annotateOpsWithBufferizationMarkers(Operation *op,
/// Rewrite pattern that bufferizes bufferizable ops.
struct BufferizationPattern
: public OpInterfaceRewritePattern<BufferizableOpInterface> {
BufferizationPattern(MLIRContext *context, BufferizationState &state,
BufferizationPattern(MLIRContext *context, const BufferizationState &state,
PatternBenefit benefit = 1)
: OpInterfaceRewritePattern<BufferizableOpInterface>(context, benefit),
state(state) {}
@ -646,9 +646,8 @@ mlir::linalg::comprehensive_bufferize::analyzeOp(Operation *op,
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<BufferizationPattern>(op->getContext(), state);
@ -665,7 +664,5 @@ LogicalResult mlir::linalg::comprehensive_bufferize::runComprehensiveBufferize(
return failure();
if (options->testAnalysisOnly)
return success();
if (failed(bufferizeOp(op, state)))
return failure();
return success();
return bufferizeOp(op, state);
}