[all-commits] [llvm/llvm-project] bc3867: [mlir] Make sure linearizeCollapsedDims doesn't dr...

[MaheshRavishankar via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: bc38673e4de50b995f4bc46d1a4b0ad95bef2356
      https://github.com/llvm/llvm-project/commit/bc38673e4de50b995f4bc46d1a4b0ad95bef2356
  Author: MaheshRavishankar <ravishankarm at google.com>
  Date:   2021-11-30 (Tue, 30 Nov 2021)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
    M mlir/test/Dialect/Linalg/reshape_linearization_fusion.mlir

  Log Message:
  -----------
  [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