mlir::xegpu Namespace Reference

Functions
void	populateXeGPUFoldAliasOpsPatterns (RewritePatternSet &patterns)
	Appends patterns for folding aliasing ops into XeGPU ops into patterns. More...
static void	transpose (llvm::ArrayRef< int64_t > trans, SmallVector< int64_t > &shape)
template<typename T >
static std::string	makeString (T array, bool breakline=false)
static SmallVector< int64_t >	getShapeOf (Type type)
static int64_t	getRankOf (Value val)
static bool	isReadHintOrNone (const CachePolicyAttr &attr)
static bool	isWriteHintOrNone (const CachePolicyAttr &attr)

Function Documentation

getRankOf()

static int64_t mlir::xegpu::getRankOf ( Value  val )

static

Definition at line 53 of file XeGPUOps.cpp.

References mlir::Value::getType().

getShapeOf()

static SmallVector<int64_t> mlir::xegpu::getShapeOf ( Type  type )

static

Definition at line 44 of file XeGPUOps.cpp.

isReadHintOrNone()

static bool mlir::xegpu::isReadHintOrNone ( const CachePolicyAttr &  attr )

static

Definition at line 60 of file XeGPUOps.cpp.

isWriteHintOrNone()

static bool mlir::xegpu::isWriteHintOrNone ( const CachePolicyAttr &  attr )

static

Definition at line 68 of file XeGPUOps.cpp.

makeString()

template<typename T >

static std::string mlir::xegpu::makeString ( T  array, bool  breakline = false  )

static

Definition at line 29 of file XeGPUOps.cpp.

populateXeGPUFoldAliasOpsPatterns()

void mlir::xegpu::populateXeGPUFoldAliasOpsPatterns

(

RewritePatternSet &

patterns

)

Appends patterns for folding aliasing ops into XeGPU ops into patterns.

Definition at line 65 of file XeGPUFoldAliasOps.cpp.

References mlir::RewritePatternSet::add(), and mlir::RewritePatternSet::getContext().

transpose()

static void mlir::xegpu::transpose ( llvm::ArrayRef< int64_t >  trans, SmallVector< int64_t > &  shape  )

static

Definition at line 21 of file XeGPUOps.cpp.

Referenced by mlir::nvgpu::getLdMatrixParams(), and pruneNonTransposedDims().