Home
last modified time | relevance | path

Searched refs:operandRole (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DNvGpuSupport.cpp70 info.operandRole = MatMulOperandRole::C; in getWarpMatrixInfo()
76 info.operandRole = MatMulOperandRole::A; in getWarpMatrixInfo()
80 info.operandRole = MatMulOperandRole::B; in getWarpMatrixInfo()
88 bool isAcc = isAccumulatorOrResult(type.operandRole); in inferTileWidthInBits()
102 const bool isAccum = isAccumulatorOrResult(type.operandRole); in getMmaSyncRegisterType()
194 isAccumulatorOrResult(fragmentType.operandRole), elementsPerRegister, in getLaneIdAndValueIdToOperandCoord()
213 if (type.operandRole == MatMulOperandRole::A || in getLdMatrixParams()
214 type.operandRole == MatMulOperandRole::C) { in getLdMatrixParams()
H A DNvGpuSupport.h33 MatMulOperandRole operandRole; member