[all-commits] [llvm/llvm-project] 9085f0: [mlir][vector] Support vector.extract distribution...

[Matthias Springer via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9085f00b4d4526013e9ae98e31cef3d1e64b92f9
      https://github.com/llvm/llvm-project/commit/9085f00b4d4526013e9ae98e31cef3d1e64b92f9
  Author: Matthias Springer <springerm at google.com>
  Date:   2023-01-09 (Mon, 09 Jan 2023)

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

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

Ops such as `%1 = vector.extract %0[2] : vector<5x96xf32>`.

Distribute the source vector, then extract. In case of a 1d extract, rewrite to vector.extractelement.

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