[all-commits] [llvm/llvm-project] a0ef8a: [mlir][Linalg] Expose vectorization precondition c...

[MaheshRavishankar via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: a0ef8af8d5d04c8a7b18cd1ce4c2a728d2f480b2
      https://github.com/llvm/llvm-project/commit/a0ef8af8d5d04c8a7b18cd1ce4c2a728d2f480b2
  Author: Mahesh Ravishankar <ravishankarm at google.com>
  Date:   2022-10-05 (Wed, 05 Oct 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp

  Log Message:
  -----------
  [mlir][Linalg] Expose vectorization precondition check as a utility function.

This patch exposes the method to check if an op can be vectorized or
not for downstream uses. Also adds a check to mark elementwise operations
that have non-vectorizable ops (like `tensor.extract`) as non vectorizable.

Reviewed By: nicolasvasilache, dcaballe, ThomasRaoux

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