Skip to content

Commit 837fb98

Browse files
authored
Update triton to 65d9862f5a9029827a7ae04439f07d7e7eadf859 (#288)
This rev of Triton uses llvm-project@8957e64a20fc7f4277565c6cfe3e555c119783ce which includes changes to bufferization interfaces.
1 parent 626a4c5 commit 837fb98

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

lib/Dialect/TritonTilingExt/IR/BufferizableOpInterfaceImpl.cpp

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,8 @@ namespace {
3636
/// Generic conversion for any DestinationStyleOpInterface on tensors.
3737
static 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

triton

Submodule triton updated 187 files

0 commit comments

Comments
 (0)