MLIR
20.0.0git
|
Collects information about a warp-level matrix operand represented by a VectorType. More...
#include "mlir/Dialect/NVGPU/Utils/MMAUtils.h"
Public Attributes | |
VectorType | vectorType |
MatMulOperandRole | operandRole |
Collects information about a warp-level matrix operand represented by a VectorType.
Definition at line 34 of file MMAUtils.h.
MatMulOperandRole mlir::nvgpu::WarpMatrixInfo::operandRole |
Definition at line 36 of file MMAUtils.h.
Referenced by mlir::nvgpu::getLaneIdAndValueIdToOperandCoord(), mlir::nvgpu::getLdMatrixParams(), mlir::nvgpu::getMmaSyncRegisterType(), mlir::nvgpu::getWarpMatrixInfo(), and mlir::nvgpu::inferTileWidthInBits().
VectorType mlir::nvgpu::WarpMatrixInfo::vectorType |
Definition at line 35 of file MMAUtils.h.
Referenced by mlir::nvgpu::getLaneIdAndValueIdToOperandCoord(), mlir::nvgpu::getLdMatrixParams(), mlir::nvgpu::getMmaSyncRegisterType(), mlir::nvgpu::getWarpMatrixInfo(), inferNumRegistersPerMatrixFragment(), and mlir::nvgpu::inferTileWidthInBits().