[all-commits] [llvm/llvm-project] 46ea07: [mlir][transform] Extend `MaskedVectorizeOp` to wo...

[Andrzej Warzyński via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 46ea07bb140ee93f68f10b83b0b6f04aa2908129
      https://github.com/llvm/llvm-project/commit/46ea07bb140ee93f68f10b83b0b6f04aa2908129
  Author: Andrzej Warzynski <andrzej.warzynski at arm.com>
  Date:   2023-09-01 (Fri, 01 Sep 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/TransformOps/LinalgTransformOps.td
    M mlir/test/Dialect/Linalg/vectorization.mlir
    M mlir/test/Dialect/Linalg/vectorize-tensor-extract.mlir

  Log Message:
  -----------
  [mlir][transform] Extend `MaskedVectorizeOp` to work for regular vectorization too

This patch extends MaskedVectorizeOp so that it can be used for
"regular" (as opposed to "masked") vectorization as well. While we can
already use VectorizeOp for "regular" vectorization, that Op will also
apply various patterns on top of vectorization. That means that at the
moment, when testing the vectorizer with VectorizeOp, we are effectively
testing "vectorization + patterns", i.e. 2 things at a time.

With these updates, you can trigger "regular" vectorization with
MaskedVectorizeOp by simply skipping the vector sizes:

  transform.structured.masked_vectorize %target : !transform.any_op

Following this change we should probably also rename this Op.

Differential Revision: https://reviews.llvm.org/D157774