[all-commits] [llvm/llvm-project] 9d51b4: [mlir][vector] Support vector.extractelement distr...

[Matthias Springer via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9d51b4e4e77691930fa837dd423c648acf4beb5e
      https://github.com/llvm/llvm-project/commit/9d51b4e4e77691930fa837dd423c648acf4beb5e
  Author: Matthias Springer <springerm at google.com>
  Date:   2022-11-10 (Thu, 10 Nov 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Vector/Transforms/VectorDistribution.h
    M mlir/lib/Dialect/Vector/Transforms/VectorDistribute.cpp
    M mlir/test/Dialect/Vector/vector-warp-distribute.mlir
    M mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp

  Log Message:
  -----------
  [mlir][vector] Support vector.extractelement distribution of 1D vectors

Ops such as `%1 = vector.extractelement %0[%pos : index] : vector<96xf32>`.

In case of an extract from a 1D vector, the source vector is distributed. The lane into which the requested position falls, extracts the element and shuffles it to all other lanes.

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