Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DNvGpuSupport.cpp174 const WarpMatrixInfo &fragmentType) { in getLaneIdAndValueIdToOperandCoord() argument
175 Type elementType = fragmentType.vectorType.getElementType(); in getLaneIdAndValueIdToOperandCoord()
176 ArrayRef<int64_t> operandShape = fragmentType.vectorType.getShape(); in getLaneIdAndValueIdToOperandCoord()
178 getMmaSyncRegisterType(fragmentType); in getLaneIdAndValueIdToOperandCoord()
185 const int64_t lineSize = inferTileWidthInBits(fragmentType); in getLaneIdAndValueIdToOperandCoord()
194 isAccumulatorOrResult(fragmentType.operandRole), elementsPerRegister, in getLaneIdAndValueIdToOperandCoord()
212 params.fragmentType = type.vectorType; in getLdMatrixParams()
243 params.fragmentType.getElementType().getIntOrFloatBitWidth()); in getLaneIdToLdMatrixMatrixCoord()
245 ArrayRef<int64_t> operandShape = params.fragmentType.getShape(); in getLaneIdToLdMatrixMatrixCoord()
H A DNvGpuSupport.h68 const WarpMatrixInfo &fragmentType);
71 VectorType fragmentType; member