XeGPUOps.cpp File Reference (original) (raw)

#include "[mlir/Dialect/Arith/Utils/Utils.h](Dialect%5F2Arith%5F2Utils%5F2Utils%5F8h%5Fsource.html)"
#include "[mlir/Dialect/GPU/IR/GPUDialect.h](GPUDialect%5F8h%5Fsource.html)"
#include "[mlir/Dialect/LLVMIR/XeVMDialect.h](XeVMDialect%5F8h%5Fsource.html)"
#include "[mlir/Dialect/Utils/IndexingUtils.h](IndexingUtils%5F8h%5Fsource.html)"
#include "[mlir/Dialect/Utils/StaticValueUtils.h](StaticValueUtils%5F8h%5Fsource.html)"
#include "[mlir/Dialect/XeGPU/IR/XeGPU.h](XeGPU%5F8h%5Fsource.html)"
#include "[mlir/IR/Builders.h](Builders%5F8h%5Fsource.html)"
#include "[mlir/IR/TypeUtilities.h](TypeUtilities%5F8h%5Fsource.html)"
#include "[mlir/Interfaces/ViewLikeInterface.h](ViewLikeInterface%5F8h%5Fsource.html)"
#include "llvm/Support/Debug.h"
#include <mlir/Dialect/XeGPU/IR/XeGPUAttrInterface.cpp.inc>
#include <mlir/Dialect/XeGPU/IR/XeGPUEnums.cpp.inc>
#include <mlir/Dialect/XeGPU/IR/XeGPU.cpp.inc>

Go to the source code of this file.

Namespaces
namespace	mlir
	Include the generated interface declarations.

Macros
#define	DEBUG_TYPE "xegpu"
#define	GET_OP_CLASSES

Functions
static bool	isSharedMemory (const MemRefType &memrefTy)
template
static std::string	makeString (T array, bool breakline=false)
static SmallVector< int64_t >	getShapeOf (Type type)
static bool	isReadHintOrNone (const CachePolicyAttr &attr)
static bool	isWriteHintOrNone (const CachePolicyAttr &attr)
static LogicalResult	isValidGatherScatterParams (Type maskTy, VectorType valueTy, TensorDescType tdescTy, function_ref< InFlightDiagnostic()> emitError)
static LogicalResult	isValidGatherScatterBufferParams (Type offsetsTy, Type maskTy, VectorType valueTy, int64_t chunkSize, function_ref< InFlightDiagnostic()> emitError)
LogicalResult	IsValidMatrixOpParams (VectorType dataTy, MemDescType mdescTy, UnitAttr subgroup_block_io, DistributeLayoutAttr layout, function_ref< InFlightDiagnostic()> emitError)
static ParseResult	parseOptionalDynamicIndexList (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &values, DenseI64ArrayAttr &integers, SmallVectorImpl< Type > *valueTypes=nullptr, AsmParser::Delimiter delimiter=AsmParser::Delimiter::Square)
static void	printOptionalDynamicIndexList (OpAsmPrinter &printer, Operation *op, OperandRange values, DenseI64ArrayAttr integers)

◆ DEBUG_TYPE

#define DEBUG_TYPE "xegpu"

◆ GET_OP_CLASSES

◆ getShapeOf()

◆ isReadHintOrNone()

bool isReadHintOrNone ( const CachePolicyAttr & attr)	static

◆ isSharedMemory()

bool isSharedMemory ( const MemRefType & memrefTy)	static

◆ isValidGatherScatterBufferParams()

◆ isValidGatherScatterParams()

◆ IsValidMatrixOpParams()

LogicalResult IsValidMatrixOpParams	(	VectorType	dataTy,
MemDescType	mdescTy,
UnitAttr	subgroup_block_io,
DistributeLayoutAttr	layout,
function_ref< InFlightDiagnostic()>	emitError )

◆ isWriteHintOrNone()

bool isWriteHintOrNone ( const CachePolicyAttr & attr)	static

◆ makeString()

template

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

◆ parseOptionalDynamicIndexList()

Definition at line 412 of file XeGPUOps.cpp.

References mlir::AsmParser::emitError(), mlir::AsmParser::getBuilder(), mlir::Builder::getDenseI64ArrayAttr(), mlir::AsmParser::getNameLoc(), mlir::AsmParser::parseColonType(), mlir::AsmParser::parseCommaSeparatedList(), mlir::AsmParser::parseInteger(), mlir::AsmParser::parseOptionalLSquare(), mlir::OpAsmParser::parseOptionalOperand(), mlir::AsmParser::parseRSquare(), mlir::AsmParser::Square, and success().