[mlir][vector] Distribute vector.insert op
authorMatthias Springer <springerm@google.com>
Mon, 9 Jan 2023 15:46:01 +0000 (16:46 +0100)
committerMatthias Springer <springerm@google.com>
Mon, 9 Jan 2023 15:50:28 +0000 (16:50 +0100)
commit1523b72946c330070c5bc55d21a58800c845db50
tree31f2f08f82b3a57c2b8955afb8050d8746d7bcaa
parent6a6f62a719b9e312133df2768b2e2d7d0b68b76f
[mlir][vector] Distribute vector.insert op

In case the distributed dim of the dest vector is also a dim of the src vector, each lane inserts a smaller part of the source vector. Otherwise, one lane inserts the entire src vector and the other lanes do nothing.

Differential Revision: https://reviews.llvm.org/D137953
mlir/lib/Dialect/Vector/Transforms/VectorDistribute.cpp
mlir/test/Dialect/Vector/vector-warp-distribute.mlir