[all-commits] [llvm/llvm-project] f12639: [mlir][Linalg] Avoid collapsing dimensions of lina...

[MaheshRavishankar via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f12639d0d674327876388cfde6b4d226359284ac
      https://github.com/llvm/llvm-project/commit/f12639d0d674327876388cfde6b4d226359284ac
  Author: Mahesh Ravishankar <ravishankarm at google.com>
  Date:   2023-05-10 (Wed, 10 May 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
    M mlir/test/Dialect/Linalg/collapse-dim.mlir

  Log Message:
  -----------
  [mlir][Linalg] Avoid collapsing dimensions of linalg op that arent foldable.

The collapsing dimensions transformation is limited to only those
cases where the sequence of dimensions are contiguous in all the
ranges of the indexing maps of the operation. Add this check before
applying the transformation.

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