#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/Dialect/GPU/Transforms/Passes.h"
#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/Dialect/Utils/IndexingUtils.h"
#include "mlir/IR/AffineExpr.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/Transforms/GreedyPatternRewriteDriver.h"
#include "mlir/Dialect/GPU/Transforms/Passes.h.inc"

Namespaces
	mlir
	Include the generated interface declarations.

Macros
#define	GEN_PASS_DEF_GPUDECOMPOSEMEMREFSPASS

Functions
static MemRefType	inferCastResultType (Value source, OpFoldResult offset)

static void	setInsertionPointToStart (OpBuilder &builder, Value val)

static bool	isInsideLaunch (Operation *op)

static std::tuple< Value, OpFoldResult, SmallVector< OpFoldResult > >	getFlatOffsetAndStrides (OpBuilder &rewriter, Location loc, Value source, ArrayRef< OpFoldResult > subOffsets, ArrayRef< OpFoldResult > subStrides={})

static Value	getFlatMemref (OpBuilder &rewriter, Location loc, Value source, ValueRange offsets)

static bool	needFlatten (Value val)

static bool	checkLayout (Value val)

Macro Definition Documentation

◆ GEN_PASS_DEF_GPUDECOMPOSEMEMREFSPASS

#define GEN_PASS_DEF_GPUDECOMPOSEMEMREFSPASS

Definition at line 24 of file DecomposeMemRefs.cpp.

Function Documentation

◆ checkLayout()

static bool checkLayout ( Value val )

static

Definition at line 121 of file DecomposeMemRefs.cpp.

References mlir::Value::getType().

◆ getFlatMemref()

static Value getFlatMemref	(	OpBuilder &	rewriter,
		Location	loc,
		Value	source,
		ValueRange	offsets
	)

static

Definition at line 105 of file DecomposeMemRefs.cpp.

References mlir::getAsOpFoldResult(), getFlatOffsetAndStrides(), and inferCastResultType().

◆ getFlatOffsetAndStrides()

static std::tuple<Value, OpFoldResult, SmallVector<OpFoldResult> > getFlatOffsetAndStrides	(	OpBuilder &	rewriter,
		Location	loc,
		Value	source,
		ArrayRef< OpFoldResult >	subOffsets,
		ArrayRef< OpFoldResult >	subStrides = `{}`
	)