[mlir][vector] Fix crash in DropInnerMostUnitDims pattern
authorThomas Raoux <thomasraoux@google.com>
Thu, 19 May 2022 17:38:04 +0000 (17:38 +0000)
committerThomas Raoux <thomasraoux@google.com>
Thu, 19 May 2022 17:38:04 +0000 (17:38 +0000)
commit4c1b65e7bc923b1ed0473cad51b5070471a201a5
tree80681ced163fdc4d9f51e0b2f3793338e564b60b
parentf2676b151d6feb16d6f259ea8bac8ef16cc08f99
[mlir][vector] Fix crash in DropInnerMostUnitDims pattern

Fix number of dimensions when incrementally replacing dimensions in
affine map.

Differential Revision: https://reviews.llvm.org/D125984
mlir/lib/Dialect/Vector/Transforms/VectorTransforms.cpp
mlir/test/Dialect/Vector/vector-transfer-collapse-inner-most-dims.mlir