Skip to content

Commit 8fd49c0

Browse files
committed
format comments
Signed-off-by: dchigarev <[email protected]>
1 parent 2a38c2c commit 8fd49c0

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

mlir/lib/Conversion/VectorToXeGPU/VectorToXeGPU.cpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -362,13 +362,12 @@ static Value computeOffsets(PatternRewriter &rewriter, OpType gatScatOp,
362362
// the collapsed dimensions. Returns the new memref value and the remaining
363363
// offsets for the last targetRank dimensions. For example:
364364
// input: %memref = memref<2x4x8x32xf32>, offsets=[%i0, %i1, %i2, %i3],
365-
// targetRank=2 output: %memref[%i0, %i1, 0, 0] -> memref<8x32xf32>, returned
366-
// offsets: [%i2, %i3]
365+
// output: %memref[%i0, %i1, 0, 0] -> memref<8x32xf32>, offsets: [%i2, %i3]
367366
static std::pair<Value, SmallVector<OpFoldResult>>
368367
convertMemrefAndOffsetsToTargetRank(PatternRewriter &rewriter, Location loc,
369368
Value memref,
370369
SmallVector<OpFoldResult> offsets,
371-
int64_t targetRank) {
370+
int64_t targetRank = 2) {
372371
auto memrefType = cast<MemRefType>(memref.getType());
373372
unsigned rank = memrefType.getRank();
374373

0 commit comments

Comments
 (0)