MLIR  16.0.0git
Enumerations | Functions
mlir::NVVM Namespace Reference

Enumerations

enum  NVVMMemorySpace { kGlobalMemorySpace = 1 , kSharedMemorySpace = 3 }
 NVVM memory space identifiers. More...
 

Functions

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. More...
 
std::unique_ptr< PasscreateOptimizeForTargetPass ()
 Creates a pass that optimizes LLVM IR for the NVVM target. More...
 

Enumeration Type Documentation

◆ NVVMMemorySpace

NVVM memory space identifiers.

Enumerator
kGlobalMemorySpace 

Global memory space identifier.

kSharedMemorySpace 

Shared memory space identifier.

Definition at line 29 of file NVVMDialect.h.

Function Documentation

◆ createOptimizeForTargetPass()

std::unique_ptr< Pass > mlir::NVVM::createOptimizeForTargetPass ( )

Creates a pass that optimizes LLVM IR for the NVVM target.

Definition at line 103 of file OptimizeForNVVM.cpp.

◆ inferMMAType()

std::pair<mlir::Type, unsigned> mlir::NVVM::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().