Skip to content

Commit 2cd73e6

Browse files
committed
fixup! [mlir][vector] Refactor createWriteOrMaskedWrite
Fix comments
1 parent 18cbf37 commit 2cd73e6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1512,13 +1512,13 @@ static SmallVector<int64_t> getTiledPackShape(linalg::PackOp packOp,
15121512
/// %res = vector.transfer_write %vectorToStore into %dest
15131513
///
15141514
/// If the leading N dimensions of the destination tensor do not match
1515-
/// `inputVecSizesForLeadingDims`, where=
1516-
/// * N = rank(`inputVecSizesForLeadingDims`)),
1515+
/// `inputVecSizesForLeadingDims` (N = rank(inputVecSizesForLeadingDims)),
15171516
/// masking is applied to ensure correctness:
15181517
///
1519-
/// %write = vector.transfer_write %vectorToStore into %dest
15201518
/// %mask = vector.create_mask(%destShape)
1521-
/// %res = vector.mask %mask { %write }
1519+
/// %res = vector.mask %mask {
1520+
/// vector.transfer_write %vectorToStore into %dest
1521+
/// }
15221522
///
15231523
/// If `useInBoundsInsteadOfMasking` is set to `true`, the `in_bounds` attribute
15241524
/// is used instead of masking:

0 commit comments

Comments
 (0)