MLIR  16.0.0git
Functions
DecomposeLinalgOps.cpp File Reference
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/Linalg/IR/Linalg.h"
+ Include dependency graph for DecomposeLinalgOps.cpp:

Go to the source code of this file.

Functions

static SmallVector< OpFoldResultgetGenericOpLoopRange (OpBuilder &b, GenericOp op)
 Helper method to compute the range of a generic op. More...
 
SmallVector< OpFoldResultpermuteValues (ArrayRef< OpFoldResult > values, AffineMap map)
 Helper method to permute the list of values based on the map. More...
 
static Value getZero (OpBuilder &b, Location loc, Type elementType)
 Get zero value for an element type. More...
 

Function Documentation

◆ getGenericOpLoopRange()

static SmallVector<OpFoldResult> getGenericOpLoopRange ( OpBuilder b,
GenericOp  op 
)
static

Helper method to compute the range of a generic op.

Definition at line 104 of file DecomposeLinalgOps.cpp.

References mlir::makeComposedFoldedMultiResultAffineApply(), and mlir::OpBuilder::setInsertionPoint().

Referenced by getZero().

◆ getZero()

static Value getZero ( OpBuilder b,
Location  loc,
Type  elementType 
)
static

◆ permuteValues()

SmallVector<OpFoldResult> permuteValues ( ArrayRef< OpFoldResult values,
AffineMap  map 
)

Helper method to permute the list of values based on the map.

Definition at line 118 of file DecomposeLinalgOps.cpp.

References mlir::detail::enumerate(), mlir::AffineMap::getResults(), and mlir::AffineMap::isPermutation().