[all-commits] [llvm/llvm-project] b40e90: [mlir][Linalg] Allow collapsing subset of the reas...

[MaheshRavishankar via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: b40e901333b903fd71f17c3314d3e40f8abde074
      https://github.com/llvm/llvm-project/commit/b40e901333b903fd71f17c3314d3e40f8abde074
  Author: Mahesh Ravishankar <ravishankarm at google.com>
  Date:   2022-04-12 (Tue, 12 Apr 2022)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/ElementwiseOpFusion.cpp
    M mlir/test/Dialect/Linalg/fuse-with-reshape-by-collapsing.mlir

  Log Message:
  -----------
  [mlir][Linalg] Allow collapsing subset of the reassociations when fusing by collapsing.

This change generalizes the fusion of `tensor.expand_shape` ->
`linalg.generic` op by collapsing to handle cases where only a subset
of the reassociations specified in the `tensor.expand_shape` are valid
to be collapsed.
The method that does the collapsing is refactored to allow it to be a
generic utility when required.

Reviewed By: gysit

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