[all-commits] [llvm/llvm-project] 4d339e: [mlir][Vector] Add pattern to reorder elementwise ...

[Andrzej Warzyński via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 4d339ec91e81ae33b0f3ea0f8a3596d99645a0e9
      https://github.com/llvm/llvm-project/commit/4d339ec91e81ae33b0f3ea0f8a3596d99645a0e9
  Author: Andrzej Warzynski <andrzej.warzynski at arm.com>
  Date:   2023-06-15 (Thu, 15 Jun 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Vector/Transforms/VectorRewritePatterns.h
    M mlir/lib/Dialect/Linalg/TransformOps/LinalgTransformOps.cpp
    M mlir/lib/Dialect/Vector/Transforms/VectorTransforms.cpp
    M mlir/test/Dialect/Linalg/vectorize-tensor-extract.mlir
    A mlir/test/Dialect/Vector/sink-vector-broadcast.mlir
    M mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp

  Log Message:
  -----------
  [mlir][Vector] Add pattern to reorder elementwise and broadcast ops

The new pattern will replace elementwise(broadcast) with
broadcast(elementwise) when safe.

This change affects tests for vectorising nD-extract. In one case
("vectorize_nd_tensor_extract_with_tensor_extract") I just trimmed the
test and only preserved the key parts (scalar and contiguous load from
the original Op). We could do the same with some other tests if that
helps maintainability.

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