Apply transformation to split the single linalg op reduction into a parallel and reduction dimension. More...

#include "mlir/Dialect/Linalg/Transforms/Transforms.h"

Public Attributes
Operation *	initOrAlloc
FillOp	fillOp
LinalgOp	splitLinalgOp
LinalgOp	resultCombiningLinalgOp

Detailed Description

Apply transformation to split the single linalg op reduction into a parallel and reduction dimension.

Then create a new linalg.generic op doing the rest of the reduction. Return the new linalg op with an extra parallel dimension or failure if the transformation didn't happen.

Example:

%r = linalg.generic {indexing_maps = [affine_map<(d0) -> (d0)>,
                                      affine_map<(d0) -> ()>],
     iterator_types = ["reduction"]}
ins(%in : tensor<32xf32>)
outs(%out : tensor<f32>) {
^bb0(%arg1: f32, %arg2: f32):
  %y = arith.addf %arg1, %arg2 : f32
  linalg.yield %y : f32
} -> tensor<f32>

To:

 %cst = arith.constant 0.000000e+00 : f32
 %0 = tensor.expand_shape %in [[0, 1]]: tensor<32xf32> into tensor<4x8xf32>
 %1 = tensor.empty [4] : tensor<4xf32>
 %2 = linalg.fill ins(%cst : f32)
                  outs(%1 : tensor<4xf32>) -> tensor<4xf32>
 %3 = linalg.generic {indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
                                       affine_map<(d0, d1) -> (d0)>],
   iterator_types = ["parallel", "reduction"]}
   ins(%0 : tensor<4x8xf32>) outs(%2 : tensor<4xf32>) {
   ^bb0(%arg3: f32, %arg5: f32):
   %5 = arith.addf %arg3, %arg4 : f32
   linalg.yield %5 : f32
 } -> tensor<4xf32>
%r = linalg.generic {indexing_maps = [affine_map<(d0) -> (d0)>,
                                      affine_map<(d0) -> ()>],
  iterator_types = ["reduction"]}
  ins(%3 : tensor<4xf32>) outs(%out : tensor<f32>) {
  ^bb0(%arg3: f32, %arg4: f32):
  %5 = arith.addf %arg3, %arg4 : f32
  linalg.yield %5 : f32
} -> tensor<f32>

Definition at line 1256 of file Transforms.h.

Member Data Documentation

◆ fillOp

FillOp mlir::linalg::SplitReductionResult::fillOp

Definition at line 1258 of file Transforms.h.

◆ initOrAlloc

Operation* mlir::linalg::SplitReductionResult::initOrAlloc

Definition at line 1257 of file Transforms.h.

◆ resultCombiningLinalgOp

LinalgOp mlir::linalg::SplitReductionResult::resultCombiningLinalgOp

Definition at line 1260 of file Transforms.h.

◆ splitLinalgOp

LinalgOp mlir::linalg::SplitReductionResult::splitLinalgOp

Definition at line 1259 of file Transforms.h.

The documentation for this struct was generated from the following file: