MLIR  17.0.0git
Macros | Functions
NVGPUDialect.cpp File Reference
#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:

Go to the source code of this file.

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