#include "mlir/Dialect/NVGPU/IR/NVGPUDialect.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/DialectImplementation.h"
#include "mlir/IR/OpImplementation.h"
#include "mlir/IR/TypeUtilities.h"
#include "mlir/IR/Verifier.h"
#include "llvm/ADT/TypeSwitch.h"
#include "mlir/Dialect/NVGPU/IR/NVGPUTypes.cpp.inc"
#include "mlir/Dialect/NVGPU/IR/NVGPU.cpp.inc"
#include "mlir/Dialect/NVGPU/IR/NVGPUDialect.cpp.inc"

Include dependency graph for NVGPUDialect.cpp:

Macros
#define	GET_TYPEDEF_LIST

#define	GET_OP_LIST

#define	GET_OP_CLASSES

#define	GET_TYPEDEF_CLASSES

Functions
static bool	isLastMemrefDimUnitStride (MemRefType type)
	Return true if the last dimension of the MemRefType has unit stride. More...

static LogicalResult	verifyMmaSyncOp (Operation *op, TypedValue< VectorType > matrixA, TypedValue< VectorType > matrixB, TypedValue< VectorType > matrixC, const std::array< int64_t, 3 > &mmaShape, bool tf32Enabled, bool sparse=false)
	Performs verification for MmaSyncOp and MmaSparseSyncOp. More...

Macro Definition Documentation

◆ GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 296 of file NVGPUDialect.cpp.

◆ GET_OP_LIST

#define GET_OP_LIST

◆ GET_TYPEDEF_CLASSES

#define GET_TYPEDEF_CLASSES

Definition at line 299 of file NVGPUDialect.cpp.

◆ GET_TYPEDEF_LIST

#define GET_TYPEDEF_LIST

Function Documentation

◆ isLastMemrefDimUnitStride()

static bool isLastMemrefDimUnitStride ( MemRefType type )

static

Return true if the last dimension of the MemRefType has unit stride.

Also return true for memrefs with no strides.

Definition at line 54 of file NVGPUDialect.cpp.

References mlir::failed(), and mlir::getStridesAndOffset().

◆ verifyMmaSyncOp()

static LogicalResult verifyMmaSyncOp	(	Operation *	op,
		TypedValue< VectorType >	matrixA,
		TypedValue< VectorType >	matrixB,
		TypedValue< VectorType >	matrixC,
		const std::array< int64_t, 3 > &	mmaShape,
		bool	tf32Enabled,
		bool	sparse = `false`
	)

static

Performs verification for MmaSyncOp and MmaSparseSyncOp.

Definition at line 100 of file NVGPUDialect.cpp.

References mlir::Operation::emitError(), mlir::Operation::emitOpError(), mlir::Type::getIntOrFloatBitWidth(), mlir::Operation::getName(), mlir::Type::isBF16(), mlir::Type::isF16(), mlir::Type::isF32(), mlir::Type::isF64(), mlir::Type::isInteger(), and mlir::success().

Macros

Functions