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

[Thomas via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 69a8a7cf2dc4f8e1213f294697f649850af06f1d
      https://github.com/llvm/llvm-project/commit/69a8a7cf2dc4f8e1213f294697f649850af06f1d
  Author: Thomas Raoux <thomasraoux 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

This reverts commit 9a844c2a9b5c09b4c35d573394a99ab860621581.