MLIR
20.0.0git
|
#include "mlir/Bytecode/BytecodeOpInterface.h"
#include "mlir/Dialect/LLVMIR/BasicPtxBuilderInterface.h"
#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"
Go to the source code of this file.
Namespaces | |
mlir | |
Include the generated interface declarations. | |
mlir::NVVM | |
Macros | |
#define | GET_ATTRDEF_CLASSES |
#define | GET_OP_CLASSES |
Enumerations | |
enum | mlir::NVVM::NVVMMemorySpace { mlir::NVVM::kGlobalMemorySpace = 1 , mlir::NVVM::kSharedMemorySpace = 3 , mlir::NVVM::kConstantMemorySpace = 4 } |
NVVM memory space identifiers. More... | |
Functions | |
std::pair< mlir::Type, unsigned > | mlir::NVVM::inferMMAType (mlir::NVVM::MMATypes type, mlir::NVVM::MMAFrag frag, int nRow, int nCol, mlir::MLIRContext *context) |
Return the element type and number of elements associated with a wmma matrix of given chracteristics. More... | |
Variables | |
constexpr int | mlir::NVVM::kSharedMemoryAlignmentBit = 128 |
#define GET_ATTRDEF_CLASSES |
Definition at line 54 of file NVVMDialect.h.
#define GET_OP_CLASSES |
Definition at line 57 of file NVVMDialect.h.