tree b5df9a7ee2f0dac885806030cd83f40fbbf37da1
parent d9403c51fd0798c45716ec5ae433aa718c874860
author MaheshRavishankar <ravishankarm@google.com> 1638340644 -0800
committer Copybara-Service <copybara-worker@google.com> 1638340842 -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

GitOrigin-RevId: bc38673e4de50b995f4bc46d1a4b0ad95bef2356