diff --git a/openmp/device/CMakeLists.txt b/openmp/device/CMakeLists.txt index 8ec9a27bd8f06..b88c8e6074a2e 100644 --- a/openmp/device/CMakeLists.txt +++ b/openmp/device/CMakeLists.txt @@ -40,18 +40,14 @@ set(src_files ) if(NOT LLVM_TARGETS_TO_BUILD OR "AMDGPU" IN_LIST LLVM_TARGETS_TO_BUILD) - set(amdbc_dirs - "${CMAKE_BINARY_DIR}/../rocm-device-libs-prefix/src/rocm-device-libs-build/lib/llvm/lib/clang/${LLVM_VERSION_MAJOR}/lib/amdgcn/bitcode" - "${CMAKE_INSTALL_PREFIX}/../../amdgcn/bitcode" - "/opt/rocm/amdgcn/bitcode") - foreach(amdbc_dir ${amdbc_dirs}) - if(EXISTS "${amdbc_dir}/ockl.bc" AND NOT _ockl_bc) - set(_ockl_bc ${amdbc_dir}/ockl.bc) - endif() - if(EXISTS "${amdbc_dir}/ocml.bc" AND NOT _ocml_bc) - set(_ocml_bc ${amdbc_dir}/ocml.bc) - endif() - endforeach() + find_package(AMDDeviceLibs REQUIRED CONFIG + HINTS ${CMAKE_BINARY_DIR}/../../tools/rocm-device-libs + ${CMAKE_BINARY_DIR}/../rocm-device-libs-prefix/src/rocm-device-libs-build + ${CMAKE_INSTALL_PREFIX} + ) + get_target_property(_ocml_bc ocml IMPORTED_LOCATION) + get_target_property(_ockl_bc ockl IMPORTED_LOCATION) + if(NOT _ockl_bc) message(FATAL_ERROR "Could not find ockl.bc") endif()