Jonathan R. Madsen
b79ce10fee
* RPATH to rocprofiler_LIBRARY_DIR for ROCm < v5.2 - until v5.2 only librocprofiler64.so was symlinked in /opt/rocm. Thus linker using SOVERSION caused issues finding librocprofiler64.so.1 * Test ROCm w/ CMAKE_INSTALL_RPATH_USE_LINK_PATH=OFF * INSTALL_RPATH_USE_LINK_PATH for omnitrace exe
98 γραμμές
3.5 KiB
CMake
98 γραμμές
3.5 KiB
CMake
# Distributed under the OSI-approved BSD 3-Clause License. See accompanying file
|
|
# Copyright.txt or https://cmake.org/licensing for details.
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
if(NOT ROCM_PATH AND NOT "$ENV{ROCM_PATH}" STREQUAL "")
|
|
set(ROCM_PATH "$ENV{ROCM_PATH}")
|
|
endif()
|
|
|
|
foreach(_DIR ${ROCmVersion_DIR} ${ROCM_PATH} /opt/rocm /opt/rocm/rocprofiler)
|
|
if(EXISTS ${_DIR})
|
|
get_filename_component(_ABS_DIR "${_DIR}" REALPATH)
|
|
list(APPEND _ROCM_ROCPROFILER_PATHS ${_ABS_DIR})
|
|
endif()
|
|
endforeach()
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
find_path(
|
|
rocprofiler_ROOT_DIR
|
|
NAMES include/rocprofiler/rocprofiler.h include/rocprofiler.h
|
|
HINTS ${_ROCM_ROCPROFILER_PATHS}
|
|
PATHS ${_ROCM_ROCPROFILER_PATHS}
|
|
PATH_SUFFIXES rocprofiler)
|
|
|
|
mark_as_advanced(rocprofiler_ROOT_DIR)
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
find_path(
|
|
rocprofiler_INCLUDE_DIR
|
|
NAMES rocprofiler.h
|
|
HINTS ${rocprofiler_ROOT_DIR} ${_ROCM_ROCPROFILER_PATHS}
|
|
PATHS ${rocprofiler_ROOT_DIR} ${_ROCM_ROCPROFILER_PATHS}
|
|
PATH_SUFFIXES include include/rocprofiler rocprofiler/include)
|
|
|
|
mark_as_advanced(rocprofiler_INCLUDE_DIR)
|
|
|
|
find_path(
|
|
rocprofiler_hsa_INCLUDE_DIR
|
|
NAMES hsa.h
|
|
HINTS ${rocprofiler_ROOT_DIR} ${_ROCM_ROCPROFILER_PATHS}
|
|
PATHS ${rocprofiler_ROOT_DIR} ${_ROCM_ROCPROFILER_PATHS}
|
|
PATH_SUFFIXES include include/hsa)
|
|
|
|
mark_as_advanced(rocprofiler_hsa_INCLUDE_DIR)
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
find_library(
|
|
rocprofiler_LIBRARY
|
|
NAMES rocprofiler64 rocprofiler
|
|
HINTS ${rocprofiler_ROOT_DIR}/rocprofiler ${rocprofiler_ROOT_DIR}
|
|
${_ROCM_ROCPROFILER_PATHS}
|
|
PATHS ${rocprofiler_ROOT_DIR}/rocprofiler ${rocprofiler_ROOT_DIR}
|
|
${_ROCM_ROCPROFILER_PATHS}
|
|
PATH_SUFFIXES lib lib64
|
|
NO_DEFAULT_PATH)
|
|
|
|
find_library(
|
|
rocprofiler_hsa-runtime_LIBRARY
|
|
NAMES hsa-runtime64 hsa-runtime
|
|
HINTS ${rocprofiler_ROOT_DIR} ${_ROCM_ROCPROFILER_PATHS}
|
|
PATHS ${rocprofiler_ROOT_DIR} ${_ROCM_ROCPROFILER_PATHS}
|
|
PATH_SUFFIXES lib lib64)
|
|
|
|
if(rocprofiler_LIBRARY)
|
|
get_filename_component(rocprofiler_LIBRARY_DIR "${rocprofiler_LIBRARY}" PATH CACHE)
|
|
endif()
|
|
|
|
mark_as_advanced(rocprofiler_LIBRARY rocprofiler_hsa-runtime_LIBRARY)
|
|
unset(_ROCM_ROCPROFILER_PATHS)
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
find_package_handle_standard_args(
|
|
rocprofiler DEFAULT_MSG rocprofiler_ROOT_DIR rocprofiler_INCLUDE_DIR
|
|
rocprofiler_hsa_INCLUDE_DIR rocprofiler_LIBRARY rocprofiler_hsa-runtime_LIBRARY)
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
if(rocprofiler_FOUND)
|
|
add_library(rocprofiler::rocprofiler INTERFACE IMPORTED)
|
|
add_library(rocprofiler::roctx INTERFACE IMPORTED)
|
|
set(rocprofiler_INCLUDE_DIRS ${rocprofiler_INCLUDE_DIR}
|
|
${rocprofiler_hsa_INCLUDE_DIR})
|
|
set(rocprofiler_LIBRARIES ${rocprofiler_LIBRARY} ${rocprofiler_hsa-runtime_LIBRARY})
|
|
set(rocprofiler_LIBRARY_DIRS ${rocprofiler_LIBRARY_DIR})
|
|
|
|
target_include_directories(
|
|
rocprofiler::rocprofiler INTERFACE ${rocprofiler_INCLUDE_DIR}
|
|
${rocprofiler_hsa_INCLUDE_DIR})
|
|
|
|
target_link_libraries(rocprofiler::rocprofiler INTERFACE ${rocprofiler_LIBRARIES})
|
|
endif()
|