tree b5df9a7ee2f0dac885806030cd83f40fbbf37da1
parent 52cf333198213e2c9c4504f560fc0161fd662191
author Thomas Raoux <thomasraoux@google.com> 1638341472 -0800
committer Copybara-Service <copybara-worker@google.com> 1638341780 -0800

[mlir] Make sure linearizeCollapsedDims doesn't drop input map dims

The new affine map generated by linearizeCollapsedDims should not drop
dimensions. We need to make sure we create a map with at least as many
dimensions as the source map. This prevents
FoldProducerReshapeOpByLinearization from generating invalid IR.

This solves regression in IREE due to https://github.com/llvm/llvm-project/commit/e4e4da86aff5606ef792d987a3ec85639219228c

Reviewed By: mravishankar

Differential Revision: https://reviews.llvm.org/D114838

This reverts commit 9a844c2a9b5c09b4c35d573394a99ab860621581.

GitOrigin-RevId: 69a8a7cf2dc4f8e1213f294697f649850af06f1d