[all-commits] [llvm/llvm-project] 9d5239: [mlir][Linalg] Add fusion of IndexedGenericOp with...

[MaheshRavishankar via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: 9d5239d39e48b8b171a0fbc47dbbb22381f4d9be
      https://github.com/llvm/llvm-project/commit/9d5239d39e48b8b171a0fbc47dbbb22381f4d9be
  Author: MaheshRavishankar <ravishankarm at google.com>
  Date:   2020-10-27 (Tue, 27 Oct 2020)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
    M mlir/test/Dialect/Linalg/reshape_fusion.mlir

  Log Message:
  -----------
  [mlir][Linalg] Add fusion of IndexedGenericOp with TensorReshapeOp by expansion.

This patch adds support for fusing linalg.indexed_generic op with
linalg.tensor_reshape op by expansion, i.e.
- linalg.indexed_generic op -> linalg.tensor_reshape op when the
  latter is expanding.
- linalg.tensor_reshape op -> linalg.indexed_generic op when the
  former is folding.

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