[all-commits] [llvm/llvm-project] b40e90: [mlir][Linalg] Allow collapsing subset of the reas...
MaheshRavishankar via All-commits
all-commits at lists.llvm.org
Tue Apr 12 11:56:50 PDT 2022
Author: Mahesh Ravishankar <ravishankarm at google.com>
Date: 2022-04-12 (Tue, 12 Apr 2022)
[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
More information about the All-commits