[mlir][vector] Fix folding of vector.extract from vector.broadcast
authorNicolas Vasilache <nicolas.vasilache@gmail.com>
Mon, 28 Nov 2022 14:12:03 +0000 (06:12 -0800)
committerNicolas Vasilache <nicolas.vasilache@gmail.com>
Mon, 28 Nov 2022 15:17:31 +0000 (07:17 -0800)
commit0650e1bcc05af64d25ad89f87b2e21b37fd88114
tree712f270eebfc4c5da51d2ee3527732f1c6826781
parent846b676766d8928249790cbb73f9fb4f1c4c8e0d
[mlir][vector] Fix folding of vector.extract from vector.broadcast

This revision fixes a bug in the vector.extract folding that was missing
handling the "dim-1" broadcasting case in vector.broadcast.

Differential Revision: https://reviews.llvm.org/D138804
mlir/include/mlir/Dialect/Vector/IR/VectorOps.td
mlir/lib/Dialect/Vector/IR/VectorOps.cpp
mlir/test/Dialect/Vector/canonicalize.mlir