[all-commits] [llvm/llvm-project] d09bef: [MLIR] Vectorize tensor.extract on 1-d tensor

[Jerry Wu via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: d09bef82c0709c5755ce33da20532481b7da2245
      https://github.com/llvm/llvm-project/commit/d09bef82c0709c5755ce33da20532481b7da2245
  Author: Che-Yu Wu <cheyuw at google.com>
  Date:   2022-10-18 (Tue, 18 Oct 2022)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/Vectorization.cpp
    M mlir/lib/Dialect/Linalg/Utils/Utils.cpp
    M mlir/test/Dialect/Linalg/vectorization.mlir

  Log Message:
  -----------
  [MLIR] Vectorize tensor.extract on 1-d tensor

This patch implements the vectorization of tensor.extract for the
basic 1-d lookup case. It only vectorizes the tensor.extract to a
vector.gather when the op extracts value from an 1-d tensor.

Related discussion: https://github.com/iree-org/iree/issues/9198

Reviewed By: dcaballe

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