Searched refs:FragmentElementInfo (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/ |
| H A D | NvGpuSupport.cpp | 99 FailureOr<FragmentElementInfo>
106 return FragmentElementInfo{ in getMmaSyncRegisterType()
115 ? FragmentElementInfo{LLVM::getFixedVectorType(f64Ty, 2), 2, 128, in getMmaSyncRegisterType()
117 : FragmentElementInfo{f64Ty, 1, 64, in getMmaSyncRegisterType()
123 return FragmentElementInfo{ in getMmaSyncRegisterType()
130 return FragmentElementInfo{ in getMmaSyncRegisterType()
137 return FragmentElementInfo{ in getMmaSyncRegisterType()
146 ? FragmentElementInfo{LLVM::getFixedVectorType(f32Ty, 2), 2, 64, in getMmaSyncRegisterType()
148 : FragmentElementInfo{f32Ty, 1, 32, in getMmaSyncRegisterType()
177 FailureOr<nvgpu::FragmentElementInfo> regInfo = in getLaneIdAndValueIdToOperandCoord()
|
| H A D | NvGpuSupport.h | 47 struct FragmentElementInfo { struct
56 FailureOr<FragmentElementInfo>
|
| H A D | VectorToGPU.cpp | 433 getMmaSyncVectorOperandType(const nvgpu::FragmentElementInfo ®Info) { in getMmaSyncVectorOperandType()
452 FailureOr<nvgpu::FragmentElementInfo> regInfo = in convertConstantOpMmaSync()
478 FailureOr<nvgpu::FragmentElementInfo> regInfo = in creatLdMatrixCompatibleLoads()
520 FailureOr<nvgpu::FragmentElementInfo> regInfo = in createNonLdMatrixLoads()
642 FailureOr<nvgpu::FragmentElementInfo> regInfo = in convertTransferWriteToStores()
|