mlir::nvgpu::WarpMatrixInfo Struct Reference

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

Detailed Description

Collects information about a warp-level matrix operand represented by a VectorType.

Definition at line 34 of file MMAUtils.h.

Member Data Documentation

operandRole

MatMulOperandRole mlir::nvgpu::WarpMatrixInfo::operandRole

Definition at line 36 of file MMAUtils.h.

Referenced by mlir::nvgpu::getLaneIdAndValueIdToOperandCoord(), mlir::nvgpu::getLdMatrixParams(), mlir::nvgpu::getMmaSyncRegisterType(), and mlir::nvgpu::inferTileWidthInBits().

vectorType

VectorType mlir::nvgpu::WarpMatrixInfo::vectorType

Definition at line 35 of file MMAUtils.h.

Referenced by mlir::nvgpu::getLaneIdAndValueIdToOperandCoord(), mlir::nvgpu::getLdMatrixParams(), mlir::nvgpu::getMmaSyncRegisterType(), inferNumRegistersPerMatrixFragment(), and mlir::nvgpu::inferTileWidthInBits().

---

The documentation for this struct was generated from the following file:

• include/mlir/Dialect/NVGPU/Utils/MMAUtils.h