MLIR  16.0.0git
Classes | Namespaces | Enumerations | Functions
NvGpuSupport.h File Reference
#include "mlir/Dialect/Affine/IR/AffineOps.h"
#include "mlir/Dialect/LLVMIR/LLVMTypes.h"
#include "mlir/Dialect/LLVMIR/NVVMDialect.h"
#include "mlir/Dialect/Utils/StructuredOpsUtils.h"
#include "mlir/Dialect/Vector/IR/VectorOps.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/IR/Types.h"
+ Include dependency graph for NvGpuSupport.h:
+ This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  mlir::nvgpu::WarpMatrixInfo
 Collects information about a warp-level matrix operand represented by a VectorType. More...
 
struct  mlir::nvgpu::FragmentElementInfo
 Specifies information about the registers which compose a matrix fragment according to the PTX documentation. More...
 
struct  mlir::nvgpu::LdMatrixParams
 
struct  mlir::nvgpu::PrepareContractToGPUMMASync
 

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::nvgpu
 

Enumerations

enum  mlir::nvgpu::MatMulOperandRole : int32_t { mlir::nvgpu::MatMulOperandRole::A = 0, mlir::nvgpu::MatMulOperandRole::B, mlir::nvgpu::MatMulOperandRole::C }
 

Functions

FailureOr< WarpMatrixInfo > mlir::nvgpu::getWarpMatrixInfo (Operation *op)
 Given an op that operates on a VectorType representing a warp-level matrix operand, the function returns a struct containing relevant type information. More...
 
int64_t mlir::nvgpu::inferTileWidthInBits (const WarpMatrixInfo &type)
 Returns the number of bits in a single tile row. More...
 
FailureOr< FragmentElementInfo > mlir::nvgpu::getMmaSyncRegisterType (const WarpMatrixInfo &type)
 Returns a FragmentElementInfo struct describing the register types for the given matrix fragment type. More...
 
FailureOr< AffineMap > mlir::nvgpu::getLaneIdAndValueIdToOperandCoord (Location loc, OpBuilder &builder, const WarpMatrixInfo &fragmentType)
 Returns an AffineMap which maps a two dimensions representing (laneId, logicalValueId) and returns two results representing offsets within a matrix operand. More...
 
FailureOr< nvgpu::LdMatrixParams > mlir::nvgpu::getLdMatrixParams (const WarpMatrixInfo &type, bool transpose)
 
FailureOr< AffineMap > mlir::nvgpu::getLaneIdToLdMatrixMatrixCoord (Location loc, OpBuilder &builder, const LdMatrixParams &params)
 Returns an AffineMap which maps a single dimension representing the laneId to two results representing offsets within the matrix operand that should be the pointer locations a thread should pass to the ldmatrix instruction. More...