diff --git a/mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td b/mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td --- a/mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td +++ b/mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td @@ -1428,6 +1428,12 @@ SSA values, the handle must be mapped to exactly one payload op with exactly one index-typed result. + Note: The input vector sizes must be bigger than or equal to their + counterpart iteration space sizes. + + Typically this operator should be applied to linalg operations that have + already be tiled to the appropriate sizes. + #### Return modes: This operation produces a definite failure if the dynamic vector sizes (SSA diff --git a/mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp b/mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp --- a/mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp +++ b/mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp @@ -987,7 +987,7 @@ return ShapedType::isDynamic(staticSize) || staticSize <= inputSize; }) && - "Input vector sizes must be smaller or equal than iteration space " + "Input vector sizes must be greater than or equal to iteration space " "static sizes"); } @@ -1050,8 +1050,8 @@ /// Emit a suitable vector form for a Linalg op. If provided, `inputVectorSizes` /// are used to vectorize this operation. `inputVectorSizes` must match the rank -/// of the iteration space of the operation and the sizes must be smaller or -/// equal than their counterpart interation space sizes, if static. +/// of the iteration space of the operation and the input vector sizes must be +/// greater than or equal to their counterpart iteration space sizes, if static. /// `inputVectorShapes` also allows the vectorization of operations with dynamic /// shapes. LogicalResult mlir::linalg::vectorize(RewriterBase &rewriter, LinalgOp linalgOp,