[all-commits] [llvm/llvm-project] c4486c: [mlir][Linalg] Fix reshape fusion to reshape the o...

MaheshRavishankar via All-commits all-commits at lists.llvm.org
Mon Jan 11 09:31:24 PST 2021


  Branch: refs/heads/master
  Home:   https://github.com/llvm/llvm-project
  Commit: c4486cfd556869a837911c7719fb6c36018bbd1f
      https://github.com/llvm/llvm-project/commit/c4486cfd556869a837911c7719fb6c36018bbd1f
  Author: MaheshRavishankar <ravishankarm at google.com>
  Date:   2021-01-11 (Mon, 11 Jan 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgOps.td
    M mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
    M mlir/test/Dialect/Linalg/reshape_fusion.mlir
    M mlir/test/Dialect/Linalg/reshape_linearization_fusion.mlir

  Log Message:
  -----------
  [mlir][Linalg] Fix reshape fusion to reshape the outs instead of creating new tensors.

When fusing tensor_reshape ops with generic/indexed_Generic op, new
linalg.init_tensor operations were created for the `outs` of the fused
op. While correct (technically) it is better to just reshape the
original `outs` operands and rely on canonicalization of init_tensor
-> tensor_reshape to achieve the same effect.

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




More information about the All-commits mailing list