[mlir][vector] Fix bug when swapping scf.for and vector warp op
authorThomas Raoux <thomasraoux@google.com>
Fri, 24 Jun 2022 19:11:36 +0000 (19:11 +0000)
committerThomas Raoux <thomasraoux@google.com>
Fri, 24 Jun 2022 19:13:02 +0000 (19:13 +0000)
commitd343cdd50992ea3137146bebf32e538b73359144
treee5dc258dc87810c9322f38b54acd35e9716ddd37
parentf1b1bcdbd4b405876577693f402d1787ca6a2985
[mlir][vector] Fix bug when swapping scf.for and vector warp op

When creating a scf.for without argument a scf.yield is automatically
created. Make sure we don't create a second one.

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