[all-commits] [llvm/llvm-project] 33d2a7: [mlir][linalg] Add pattern to split reduction dime...

[Thomas via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 33d2a780a1397cb0203692019fc95f0784eff0fe
      https://github.com/llvm/llvm-project/commit/33d2a780a1397cb0203692019fc95f0784eff0fe
  Author: Thomas Raoux <thomasraoux at google.com>
  Date:   2022-03-24 (Thu, 24 Mar 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/Linalg/IR/LinalgInterfaces.td
    M mlir/include/mlir/Dialect/Linalg/Transforms/Transforms.h
    M mlir/lib/Dialect/Linalg/Transforms/CMakeLists.txt
    A mlir/lib/Dialect/Linalg/Transforms/SplitReduction.cpp
    A mlir/test/Dialect/Linalg/split_reduction.mlir
    M mlir/test/lib/Dialect/Linalg/TestLinalgTransforms.cpp

  Log Message:
  -----------
  [mlir][linalg] Add pattern to split reduction dimension in a linalg op

This transformation allow to break up a reduction dimension in a
parallel and a reduction dimension. This is followed by a separate
reduction op. This allows to generate tree reduction which is beneficial
on target allowing to take advantage parallelism.

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