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 /// Bufferize `op` and its nested ops. Bufferization decisions are stored in
/// `state`. /// `state`.
LogicalResult bufferizeOp(Operation *op, BufferizationState &state); LogicalResult bufferizeOp(Operation *op, const BufferizationState &state);
/// Run Comprehensive Bufferize on the given op: Analysis + Bufferization /// Run Comprehensive Bufferize on the given op: Analysis + Bufferization
LogicalResult LogicalResult

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

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