[all-commits] [llvm/llvm-project] 0ccc44: [mlir][linalg] Fix tile and fuse for outermost red...

[Tobias Gysi via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0ccc44cec067abbc702d5d3afb44e0395c55820d
      https://github.com/llvm/llvm-project/commit/0ccc44cec067abbc702d5d3afb44e0395c55820d
  Author: Tobias Gysi <gysit at google.com>
  Date:   2021-11-22 (Mon, 22 Nov 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
    M mlir/lib/Dialect/Linalg/Transforms/FusionOnTensors.cpp
    M mlir/test/Dialect/Linalg/tile-and-fuse-on-tensors.mlir

  Log Message:
  -----------
  [mlir][linalg] Fix tile and fuse for outermost reduction.

Tile and fuse failed if the outermost tile loop is a reduction dimension. Add the necessary check to handle outermost reductions and introduce a test case to verify the change.

Reviewed By: nicolasvasilache

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