[mlir][vector] Extend vector unroll to all element-wise ops
authorThomas Raoux <thomasraoux@google.com>
Mon, 21 Dec 2020 21:17:17 +0000 (13:17 -0800)
committerThomas Raoux <thomasraoux@google.com>
Mon, 21 Dec 2020 21:31:22 +0000 (13:31 -0800)
commit7c7b55b985136a975223a9cefccd8fa1a5df7765
tree8ae717583f9feaa732bdc1d0867bcb4558cf431c
parent9d2529a38b34d06dbe17020b98db1ee21d9a628c
[mlir][vector] Extend vector unroll to all element-wise ops

Extend unroll to support all element-wise ops and allow unrolling for ops with
vector operands of with the same shape as the destination but different element
type (like Cmp or Select).

Differential Revision: https://reviews.llvm.org/D93121
mlir/include/mlir/Dialect/StandardOps/IR/Ops.td
mlir/lib/Dialect/Vector/VectorTransforms.cpp
mlir/test/Dialect/Vector/vector-transforms.mlir
mlir/test/lib/Transforms/TestVectorTransforms.cpp