[mlir][Vector] Support broadcast vector type in distribution of vector.warp_execute_...
authorNicolas Vasilache <nicolas.vasilache@gmail.com>
Tue, 13 Sep 2022 07:51:56 +0000 (00:51 -0700)
committerNicolas Vasilache <nicolas.vasilache@gmail.com>
Tue, 13 Sep 2022 15:18:47 +0000 (08:18 -0700)
commit845dc178c0bb36af76229f89da7e13e866b010cd
treea68497a2b2327767fbe11504de2c513b54e7e155
parentb8266f512a6218e8319379b65a637fa82b3900f6
[mlir][Vector] Support broadcast vector type in  distribution of vector.warp_execute_on_lane_0.

This revision significantly improves and tests the broadcast behavior of vector.warp_execute_on_lane_0.

Previously, the implementation of the broadcast behavior of vector.warp_execute_on_lane_0
assumed that the broadcasted value was always of scalar type.

This is not necessarily the case.

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