Lines Matching refs:regInfo

433 getMmaSyncVectorOperandType(const nvgpu::FragmentElementInfo &regInfo) {  in getMmaSyncVectorOperandType()  argument
434 SmallVector<int64_t> shape{regInfo.numRegistersPerFragment, in getMmaSyncVectorOperandType()
435 regInfo.elementsPerRegister}; in getMmaSyncVectorOperandType()
436 Type elType = regInfo.registerLLVMType; in getMmaSyncVectorOperandType()
452 FailureOr<nvgpu::FragmentElementInfo> regInfo = in convertConstantOpMmaSync() local
454 if (failed(regInfo)) in convertConstantOpMmaSync()
457 VectorType vectorType = getMmaSyncVectorOperandType(*regInfo); in convertConstantOpMmaSync()
478 FailureOr<nvgpu::FragmentElementInfo> regInfo = in creatLdMatrixCompatibleLoads() local
480 if (failed(regInfo)) in creatLdMatrixCompatibleLoads()
500 VectorType vectorType = getMmaSyncVectorOperandType(*regInfo); in creatLdMatrixCompatibleLoads()
520 FailureOr<nvgpu::FragmentElementInfo> regInfo = in createNonLdMatrixLoads() local
522 if (failed(regInfo)) { in createNonLdMatrixLoads()
532 Type loadedElType = regInfo->registerLLVMType; in createNonLdMatrixLoads()
533 VectorType vectorType = getMmaSyncVectorOperandType(*regInfo); in createNonLdMatrixLoads()
556 builder.getIndexAttr(i * regInfo->elementsPerRegister)); in createNonLdMatrixLoads()
576 builder.getIndexAttr(i * regInfo->elementsPerRegister + innerIdx)); in createNonLdMatrixLoads()
642 FailureOr<nvgpu::FragmentElementInfo> regInfo = in convertTransferWriteToStores() local
644 if (failed(regInfo)) in convertTransferWriteToStores()
647 VectorType vectorType = getMmaSyncVectorOperandType(*regInfo); in convertTransferWriteToStores()
653 b.getIndexAttr(i * regInfo->elementsPerRegister)); in convertTransferWriteToStores()