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::getWarpMatrixInfo().

vectorType

VectorType mlir::nvgpu::WarpMatrixInfo::vectorType

Definition at line 35 of file MMAUtils.h.

Referenced by mlir::nvgpu::getWarpMatrixInfo(), and inferNumRegistersPerMatrixFragment().

---

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

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