MLIR  14.0.0git
Macros | Functions
NVVMDialect.h File Reference
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/IR/Dialect.h"
#include "mlir/IR/OpDefinition.h"
#include "mlir/Interfaces/SideEffectInterfaces.h"
#include "llvm/IR/IntrinsicsNVPTX.h"
#include "mlir/Dialect/LLVMIR/NVVMOpsEnums.h.inc"
#include "mlir/Dialect/LLVMIR/NVVMOpsAttributes.h.inc"
#include "mlir/Dialect/LLVMIR/NVVMOps.h.inc"
#include "mlir/Dialect/LLVMIR/NVVMOpsDialect.h.inc"
Include dependency graph for NVVMDialect.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros

#define GET_ATTRDEF_CLASSES
 
#define GET_OP_CLASSES
 

Functions

std::pair< mlir::Type, unsignedinferMMAType (mlir::NVVM::MMATypes type, mlir::NVVM::MMAFrag frag, mlir::MLIRContext *context)
 Return the element type and number of elements associated with a wmma matrix of given chracteristics. More...
 

Macro Definition Documentation

◆ GET_ATTRDEF_CLASSES

#define GET_ATTRDEF_CLASSES

Definition at line 33 of file NVVMDialect.h.

◆ GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 36 of file NVVMDialect.h.

Function Documentation

◆ inferMMAType()

std::pair<mlir::Type, unsigned> inferMMAType ( mlir::NVVM::MMATypes  type,
mlir::NVVM::MMAFrag  frag,
mlir::MLIRContext context 
)

Return the element type and number of elements associated with a wmma matrix of given chracteristics.

This matches the logic in IntrinsicsNVVM.td WMMA_REGS structure.

Referenced by mlir::convertMMAToLLVMType().