Skip to content

Commit 1107149

Browse files
committed
.
1 parent 1384e0a commit 1107149

2 files changed

Lines changed: 3 additions & 5 deletions

File tree

include/triton-shared/Conversion/TritonArithToLinalg/ConversionPatterns.hpp

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2210,11 +2210,9 @@ struct GatherConverter : public OpConversionPattern<triton::GatherOp> {
22102210
auto resultType = cast<RankedTensorType>(op.getType());
22112211
int64_t rank = resultType.getRank();
22122212

2213-
auto empty = rewriter
2213+
Value empty = rewriter
22142214
.create<tensor::EmptyOp>(loc, resultType.getShape(),
2215-
resultType.getElementType())
2216-
.getResult();
2217-
2215+
resultType.getElementType());
22182216
SmallVector<AffineMap, 2> affineMaps(2,
22192217
rewriter.getMultiDimIdentityMap(rank));
22202218
SmallVector<utils::IteratorType> iteratorTypes(

test/Conversion/TritonToLinalg/gather.mlir

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,4 +68,4 @@ module {
6868
// CHECK: %[[VAL_27:.*]] = memref.reinterpret_cast %[[VAL_2]] to offset: [0], sizes: [8, 4], strides: {{\[}}%[[VAL_10]], 1] : memref<*xf32> to memref<8x4xf32, strided<[?, 1]>>
6969
// CHECK: bufferization.materialize_in_destination %[[VAL_18]] in writable %[[VAL_27]] : (tensor<8x4xf32>, memref<8x4xf32, strided<[?, 1]>>) -> ()
7070
// CHECK: return
71-
// CHECK: }
71+
// CHECK: }

0 commit comments

Comments
 (0)