@@ -36,7 +36,8 @@ namespace {
3636// / Generic conversion for any DestinationStyleOpInterface on tensors.
3737static LogicalResult bufferizeTritonTilingExtDestinationStyleOpInterface (
3838 RewriterBase &rewriter, DestinationStyleOpInterface op,
39- const BufferizationOptions &options) {
39+ const BufferizationOptions &options,
40+ BufferizationState &state) {
4041 // Take a guard before anything else.
4142 OpBuilder::InsertionGuard g (rewriter);
4243 rewriter.setInsertionPoint (op);
@@ -58,7 +59,7 @@ static LogicalResult bufferizeTritonTilingExtDestinationStyleOpInterface(
5859 newInputBuffers.push_back (opOperand->get ());
5960 continue ;
6061 }
61- FailureOr<Value> buffer = getBuffer (rewriter, opOperand->get (), options);
62+ FailureOr<Value> buffer = getBuffer (rewriter, opOperand->get (), options, state );
6263 if (failed (buffer))
6364 return failure ();
6465 newInputBuffers.push_back (*buffer);
@@ -69,7 +70,7 @@ static LogicalResult bufferizeTritonTilingExtDestinationStyleOpInterface(
6970 for (OpResult opResult : op->getOpResults ()) {
7071 OpOperand *opOperand = op.getDpsInitOperand (opResult.getResultNumber ());
7172 FailureOr<Value> resultBuffer =
72- getBuffer (rewriter, opOperand->get (), options);
73+ getBuffer (rewriter, opOperand->get (), options, state );
7374 if (failed (resultBuffer))
7475 return failure ();
7576 newOutputBuffers.push_back (*resultBuffer);
@@ -109,9 +110,10 @@ struct TritonTilingExtOpInterface
109110 }
110111
111112 LogicalResult bufferize (Operation *op, RewriterBase &rewriter,
112- const BufferizationOptions &options) const {
113+ const BufferizationOptions &options,
114+ BufferizationState &state) const {
113115 return bufferizeTritonTilingExtDestinationStyleOpInterface (
114- rewriter, cast<DestinationStyleOpInterface>(op), options);
116+ rewriter, cast<DestinationStyleOpInterface>(op), options, state );
115117 }
116118};
117119
0 commit comments