MemRefUtils.h File Reference

#include "mlir/Dialect/MemRef/IR/MemRef.h"

Go to the source code of this file.

Classes
struct	mlir::memref::LinearizedMemRefInfo
	For a memref with offset, sizes and strides, returns the offset, size, and potentially the size padded at the front to use for the linearized memref. More...

Namespaces
namespace	mlir
	Include the generated interface declarations.
namespace	mlir::memref

Typedefs
using	mlir::MemrefValue = TypedValue<BaseMemRefType>
	A value with a memref type.

Functions
bool	mlir::memref::isStaticShapeAndContiguousRowMajor (MemRefType type)
	Returns true, if the memref type has static shapes and represents a contiguous chunk of memory.
std::pair< LinearizedMemRefInfo, OpFoldResult >	mlir::memref::getLinearizedMemRefOffsetAndSize (OpBuilder &builder, Location loc, int srcBits, int dstBits, OpFoldResult offset, ArrayRef< OpFoldResult > sizes, ArrayRef< OpFoldResult > strides, ArrayRef< OpFoldResult > indices={})
LinearizedMemRefInfo	mlir::memref::getLinearizedMemRefOffsetAndSize (OpBuilder &builder, Location loc, int srcBits, int dstBits, OpFoldResult offset, ArrayRef< OpFoldResult > sizes)
	For a memref with offset and sizes, returns the offset and size to use for the linearized memref, assuming that the strides are computed from a row-major ordering of the sizes;.
void	mlir::memref::eraseDeadAllocAndStores (RewriterBase &rewriter, Operation *parentOp)
	Track temporary allocations that are never read from.
SmallVector< OpFoldResult >	mlir::memref::computeSuffixProductIRBlock (Location loc, OpBuilder &builder, ArrayRef< OpFoldResult > sizes)
	Given a set of sizes, return the suffix product.
SmallVector< OpFoldResult >	mlir::memref::computeStridesIRBlock (Location loc, OpBuilder &builder, ArrayRef< OpFoldResult > sizes)
MemrefValue	mlir::memref::skipFullyAliasingOperations (MemrefValue source)
	Walk up the source chain until an operation that changes/defines the view of memory is found (i.e.
bool	mlir::memref::isSameViewOrTrivialAlias (MemrefValue a, MemrefValue b)
	Checks if two (memref) values are the same or statically known to alias the same region of memory.
MemrefValue	mlir::memref::skipViewLikeOps (MemrefValue source)
	Walk up the source chain until we find an operation that is not a view of the source memref (i.e.
LogicalResult	mlir::memref::resolveSourceIndicesExpandShape (Location loc, PatternRewriter &rewriter, memref::ExpandShapeOp expandShapeOp, ValueRange indices, SmallVectorImpl< Value > &sourceIndices, bool startsInbounds)
	Given the 'indices' of a load/store operation where the memref is a result of a expand_shape op, returns the indices w.r.t to the source memref of the expand_shape op.
LogicalResult	mlir::memref::resolveSourceIndicesCollapseShape (Location loc, PatternRewriter &rewriter, memref::CollapseShapeOp collapseShapeOp, ValueRange indices, SmallVectorImpl< Value > &sourceIndices)
	Given the 'indices' of a load/store operation where the memref is a result of a collapse_shape op, returns the indices w.r.t to the source memref of the collapse_shape op.