GPUDialect.h File Reference

#include "mlir/Bytecode/BytecodeOpInterface.h"
#include "mlir/Dialect/DLTI/Traits.h"
#include "mlir/Dialect/GPU/IR/CompilationInterfaces.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Dialect.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/IR/OpImplementation.h"
#include "mlir/IR/RegionKindInterface.h"
#include "mlir/IR/SymbolTable.h"
#include "mlir/Interfaces/ControlFlowInterfaces.h"
#include "mlir/Interfaces/FunctionInterfaces.h"
#include "mlir/Interfaces/InferIntRangeInterface.h"
#include "mlir/Interfaces/InferTypeOpInterface.h"
#include "mlir/Interfaces/SideEffectInterfaces.h"
#include "llvm/ADT/STLExtras.h"
#include "mlir/Dialect/GPU/IR/GPUOpsEnums.h.inc"
#include "mlir/Dialect/GPU/IR/GPUOpsDialect.h.inc"
#include "mlir/Dialect/GPU/IR/GPUOpInterfaces.h.inc"
#include "mlir/Dialect/SCF/IR/DeviceMappingInterface.h"
#include "mlir/Dialect/GPU/IR/GPUOpsAttributes.h.inc"
#include "mlir/Dialect/GPU/IR/GPUOps.h.inc"

Go to the source code of this file.

Classes
struct	mlir::gpu::KernelDim3
	Utility class for the GPU dialect to represent triples of Values accessible through .x, .y, and .z similarly to CUDA notation. More...
class	mlir::gpu::AsyncTokenType
struct	mlir::gpu::MMAMatrixStorageType
	MMAMatrixType storage and uniquing. More...
class	mlir::gpu::MMAMatrixType
	MMAMatrix represents a matrix held by a subgroup for matrix-matrix multiply accumulate operations. More...
class	mlir::gpu::SparseDnTensorHandleType
class	mlir::gpu::SparseSpMatHandleType
class	mlir::gpu::SparseSpGEMMOpHandleType

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

Macros
#define	GET_ATTRDEF_CLASSES
#define	GET_OP_CLASSES

Enumerations
enum class	mlir::gpu::SparseHandleKind { mlir::gpu::SpMat , mlir::gpu::DnTensor , mlir::gpu::SpGEMMOp }

Functions
void	mlir::gpu::addAsyncDependency (Operation *op, Value token)

Macro Definition Documentation

GET_ATTRDEF_CLASSES

#define GET_ATTRDEF_CLASSES

Definition at line 218 of file GPUDialect.h.

GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 221 of file GPUDialect.h.