MLIR
17.0.0git
|
#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"
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... | |
#define GET_OP_CLASSES |
Definition at line 296 of file NVGPUDialect.cpp.
#define GET_OP_LIST |
#define GET_TYPEDEF_CLASSES |
Definition at line 299 of file NVGPUDialect.cpp.
#define GET_TYPEDEF_LIST |
|
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().
|
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().