1if (MLIR_CUDA_CONVERSIONS_ENABLED) 2 set(NVPTX_LIBS 3 NVPTXCodeGen 4 NVPTXDesc 5 NVPTXInfo 6 ) 7endif() 8 9if (MLIR_ROCM_CONVERSIONS_ENABLED) 10 set(AMDGPU_LIBS 11 AMDGPUCodeGen 12 AMDGPUDesc 13 AMDGPUInfo 14 ) 15endif() 16 17add_mlir_conversion_library(MLIRGPUToGPURuntimeTransforms 18 ConvertLaunchFuncToRuntimeCalls.cpp 19 ConvertKernelFuncToBlob.cpp 20 21 DEPENDS 22 MLIRConversionPassIncGen 23 intrinsics_gen 24 25 LINK_COMPONENTS 26 Core 27 MC 28 ${AMDGPU_LIBS} 29 ${NVPTX_LIBS} 30 31 LINK_LIBS PUBLIC 32 MLIRGPU 33 MLIRIR 34 MLIRLLVMIR 35 MLIRPass 36 MLIRSupport 37 MLIRStandardToLLVM 38) 39