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)
static bool	isEvenDistributed (llvm::ArrayRef< int64_t > shape, xegpu::LayoutAttr attr)
static LogicalResult	isValidGatherScatterParams (Type maskTy, VectorType valueTy, TensorDescType tdescTy, UnitAttr transposeAttr, function_ref< InFlightDiagnostic()> emitError)

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.

Referenced by isValidGatherScatterParams().

isEvenDistributed()

static bool mlir::xegpu::isEvenDistributed ( llvm::ArrayRef< int64_t >  shape, xegpu::LayoutAttr  attr  )

static

Definition at line 82 of file XeGPUOps.cpp.

isReadHintOrNone()

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

static

Definition at line 60 of file XeGPUOps.cpp.

References kind.

isValidGatherScatterParams()

static LogicalResult mlir::xegpu::isValidGatherScatterParams ( Type  maskTy, VectorType  valueTy, TensorDescType  tdescTy, UnitAttr  transposeAttr, function_ref< InFlightDiagnostic()>  emitError  )

static

Definition at line 105 of file XeGPUOps.cpp.

References mlir::emitError(), getShapeOf(), makeString(), and transpose().

isWriteHintOrNone()

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

static

Definition at line 68 of file XeGPUOps.cpp.

References kind.

makeString()

template<typename T >

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

static

Definition at line 30 of file XeGPUOps.cpp.

Referenced by isValidGatherScatterParams().

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::patterns.

transpose()

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

static

Definition at line 22 of file XeGPUOps.cpp.

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