#include "mlir/Dialect/NVGPU/IR/NVGPUDialect.h"
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/DialectImplementation.h"
#include "mlir/IR/OpImplementation.h"
#include "mlir/IR/TypeUtilities.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...

Macro Definition Documentation

◆ GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 245 of file NVGPUDialect.cpp.

◆ GET_OP_LIST

#define GET_OP_LIST

◆ GET_TYPEDEF_CLASSES

#define GET_TYPEDEF_CLASSES

Definition at line 248 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 41 of file NVGPUDialect.cpp.

References mlir::Type::cast(), mlir::emitError(), mlir::failed(), mlir::Type::getIntOrFloatBitWidth(), mlir::getStridesAndOffset(), mlir::Value::getType(), mlir::Type::isBF16(), mlir::Type::isF16(), mlir::Type::isF32(), mlir::Type::isF64(), mlir::Type::isInteger(), mlir::success(), and mlir::verify().

Macros