[all-commits] [llvm/llvm-project] dddf6a: Simplifying the SplitReduction logic that uses the...

[Murali Vijayaraghavan via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: dddf6ab27212d9813a360eb95440c61e81a308be
      https://github.com/llvm/llvm-project/commit/dddf6ab27212d9813a360eb95440c61e81a308be
  Author: Murali Vijayaraghavan <muralivi at google.com>
  Date:   2022-11-17 (Thu, 17 Nov 2022)

  Changed paths:
    M mlir/lib/Dialect/Linalg/Transforms/SplitReduction.cpp
    M mlir/test/Dialect/Linalg/transform-op-split-reduction.mlir

  Log Message:
  -----------
  Simplifying the SplitReduction logic that uses the control to get the
dimension where the extra parallel dimension is inserted

Currently, the innerParallel and non innerParallel strategies use two
different ways to fix for where the extra loop is inserted and where the
extra dimension for the intermediate result is inserted - innerParallel
adds the extra (parallel) loop right after the pre-existing reduction
loop, whereas non innerParallel adds the reduction loop in the successor
to the index supplied by control, and the parallel loop in the index
supplied by the control. The semantics of the index supplied by the
control is supposed to only control where the extra tensor dimension is
inserted in the intermediate tensor. Conflating this index with where
the reduction (and parallel) loops are inserted leads to more complex
(and confusing) logic overall. This differential removes conflating the
two uses of the index, and keeps the reduction and parallel loops in the
same vicinity and uses the supplied index to only determine the position
of the extra tensor dimension. It also simplifies the code by merging
the two strategies in a lot more places.

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