[all-commits] [llvm/llvm-project] cc11ce: [mlir][Linalg] Add support for fusion between inde...

[Han-Chung Wang via All-commits]

  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: cc11ceda165b5ba0a87e812fbd6ed1bce4fefd2f
      https://github.com/llvm/llvm-project/commit/cc11ceda165b5ba0a87e812fbd6ed1bce4fefd2f
  Author: Hanhan Wang <hanchung at google.com>
  Date:   2020-06-03 (Wed, 03 Jun 2020)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/Fusion.cpp
    M mlir/test/Dialect/Linalg/fusion-tensor.mlir

  Log Message:
  -----------
  [mlir][Linalg] Add support for fusion between indexed_generic ops and generic ops on tensors.

Summary:
Different from the fusion between generic ops, indices are involved. In this
context, we need to re-map the indices for producer since the fused op is built
on consumer's perspective. This patch supports all combination of the fusion
between indexed_generic ops and generic ops, which includes tests case:
  1) generic op as producer and indexed_generic op as consumer.
  2) indexed_generic op as producer and generic op as consumer.
  3) indexed_generic op as producer and indexed_generic op as consumer.

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