[all-commits] [llvm/llvm-project] 892fdc: [mlir][Linalg] Generalize the logic to compute rea...

[MaheshRavishankar via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: 892fdc923f06adbef507ebe594fa7b48224d93f0
      https://github.com/llvm/llvm-project/commit/892fdc923f06adbef507ebe594fa7b48224d93f0
  Author: Mahesh Ravishankar <ravishankarm at google.com>
  Date:   2020-09-30 (Wed, 30 Sep 2020)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/DropUnitDims.cpp
    M mlir/test/Dialect/Linalg/drop-unit-extent-dims.mlir

  Log Message:
  -----------
  [mlir][Linalg] Generalize the logic to compute reassociation maps
while folding tensor_reshape op.

While folding reshapes that introduce unit extent dims, the logic to
compute the reassociation maps can be generalized to handle some
corner cases, for example, when the folded shape still has unit-extent
dims but corresponds to folded unit extent dims of the expanded shape.

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