Skip to content

Commit fdb0540

Browse files
committed
Add in-code comments
Signed-off-by: dchigarev <[email protected]>
1 parent 76c8129 commit fdb0540

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

mlir/lib/Conversion/VectorToXeGPU/VectorToXeGPU.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -388,6 +388,7 @@ static Value computeOffsets(PatternRewriter &rewriter, OpType gatScatOp,
388388
arith::AddIOp::create(rewriter, loc, baseOffset, offsetContrib);
389389
}
390390
Value indices = gatScatOp.getIndices();
391+
// Extract indices layout and propagate it to all 'vector' ops created here
391392
auto indicesLayout = mlir::xegpu::getDistributeLayoutAttr(indices);
392393
VectorType vecType = cast<VectorType>(indices.getType());
393394

0 commit comments

Comments
 (0)