MLIR
20.0.0git
|
Encapsulates the parameters needed to lower a nvgpu.ldmatrix
operation to nvvm.ldmatrix
.
More...
#include "mlir/Dialect/NVGPU/Utils/MMAUtils.h"
Public Attributes | |
VectorType | fragmentType |
bool | isAccum |
int64_t | numTiles |
vector::IteratorType | contiguousDimType |
NVVM::MMALayout | targetLayout |
Encapsulates the parameters needed to lower a nvgpu.ldmatrix
operation to nvvm.ldmatrix
.
Definition at line 77 of file MMAUtils.h.
vector::IteratorType mlir::nvgpu::LdMatrixParams::contiguousDimType |
Definition at line 81 of file MMAUtils.h.
Referenced by mlir::nvgpu::getLaneIdToLdMatrixMatrixCoord(), and mlir::nvgpu::getLdMatrixParams().
VectorType mlir::nvgpu::LdMatrixParams::fragmentType |
Definition at line 78 of file MMAUtils.h.
Referenced by mlir::nvgpu::getLaneIdToLdMatrixMatrixCoord(), and mlir::nvgpu::getLdMatrixParams().
bool mlir::nvgpu::LdMatrixParams::isAccum |
Definition at line 79 of file MMAUtils.h.
int64_t mlir::nvgpu::LdMatrixParams::numTiles |
Definition at line 80 of file MMAUtils.h.
Referenced by mlir::nvgpu::getLdMatrixParams().
NVVM::MMALayout mlir::nvgpu::LdMatrixParams::targetLayout |
Definition at line 82 of file MMAUtils.h.
Referenced by mlir::nvgpu::getLdMatrixParams().