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 "llvm/ADT/SmallVectorExtras.h"
#include <optional>

Go to the source code of this file.

Functions
static SmallVector< OpFoldResult >	getGenericOpLoopRange (OpBuilder &b, GenericOp op)
	Helper method to compute the range of a generic op.
static Value	getZero (OpBuilder &b, Location loc, Type elementType)
	Get zero value for an element type.

Function Documentation

getGenericOpLoopRange()

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

static

Helper method to compute the range of a generic op.

Definition at line 106 of file DecomposeLinalgOps.cpp.

References b, and mlir::affine::makeComposedFoldedMultiResultAffineApply().

getZero()

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

static

Get zero value for an element type.

Definition at line 120 of file DecomposeLinalgOps.cpp.

References b, mlir::arith::ConstantFloatOp::create(), mlir::arith::ConstantIndexOp::create(), mlir::arith::ConstantIntOp::create(), mlir::Type::isIndex(), and mlir::Type::isIntOrIndexOrFloat().