Skip to content

Commit 8a06fd5

Browse files
committed
fixup! fixup! fixup! fixup! fixup! [mlir][linalg] Enable scalable vectorization of linalg.unpack (WIP)
Fix comment
1 parent 771a1f1 commit 8a06fd5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1973,7 +1973,7 @@ vectorizeAsTensorUnpackOp(RewriterBase &rewriter, linalg::UnPackOp unpackOp,
19731973
// 1.2 Infer vector sizes for the read operation.
19741974
//
19751975
// The steps are:
1976-
// 1. readVectorSizes = vectorInputSizes
1976+
// 1. readVectorSizes = writeVectorSizes
19771977
// 2. Take readVectorSizes from 1. and divide all locations pointed by
19781978
// the inner_dims_pos attribyte by the `inner_tiles` attribute value.
19791979
// 3. If outer_dims_perms is present, permutate readVectorSizes accordingly.

0 commit comments

Comments
 (0)