Jonathan R. Madsen
114 line
4.3 KiB
CMake
114 line
4.3 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)
|
|
|
|
#----------------------------------------------------------------------------------------#
|
|
|
|
set(_ROCM_PATHS $ENV{ROCM_HOME} /opt/rocm /opt/rocm/roctracer)
|
|
|
|
#----------------------------------------------------------------------------------------#
|
|
|
|
find_path(roctracer_ROOT_DIR
|
|
NAMES include/roctracer.h
|
|
HINTS ${_ROCM_PATHS}
|
|
PATHS ${_ROCM_PATHS}
|
|
PATH_SUFFIXES roctracer)
|
|
|
|
mark_as_advanced(roctracer_ROOT_DIR)
|
|
|
|
#----------------------------------------------------------------------------------------#
|
|
|
|
find_path(roctracer_INCLUDE_DIR
|
|
NAMES roctracer.h
|
|
HINTS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATHS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATH_SUFFIXES roctracer/include include)
|
|
|
|
mark_as_advanced(roctracer_INCLUDE_DIR)
|
|
|
|
find_path(roctracer_hsa_INCLUDE_DIR
|
|
NAMES hsa.h
|
|
HINTS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATHS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATH_SUFFIXES include include/hsa)
|
|
|
|
mark_as_advanced(roctracer_hsa_INCLUDE_DIR)
|
|
|
|
#----------------------------------------------------------------------------------------#
|
|
|
|
find_library(roctracer_LIBRARY
|
|
NAMES roctracer64 roctracer
|
|
HINTS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATHS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATH_SUFFIXES lib lib64)
|
|
|
|
find_library(roctracer_roctx_LIBRARY
|
|
NAMES roctx64 roctx
|
|
HINTS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATHS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATH_SUFFIXES lib lib64)
|
|
|
|
find_library(roctracer_kfdwrapper_LIBRARY
|
|
NAMES kfdwrapper64 kfdwrapper
|
|
HINTS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATHS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATH_SUFFIXES lib lib64)
|
|
|
|
find_library(roctracer_hsakmt_LIBRARY
|
|
NAMES hsakmt
|
|
HINTS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATHS ${roctracer_ROOT_DIR} ${_ROCM_PATHS}
|
|
PATH_SUFFIXES lib lib64)
|
|
|
|
if(roctracer_LIBRARY)
|
|
get_filename_component(roctracer_LIBRARY_DIR "${roctracer_LIBRARY}" PATH CACHE)
|
|
endif()
|
|
|
|
mark_as_advanced(roctracer_LIBRARY roctracer_roctx_LIBRARY)
|
|
|
|
#----------------------------------------------------------------------------------------#
|
|
|
|
find_package_handle_standard_args(roctracer DEFAULT_MSG
|
|
roctracer_ROOT_DIR
|
|
roctracer_INCLUDE_DIR
|
|
roctracer_hsa_INCLUDE_DIR
|
|
roctracer_LIBRARY
|
|
roctracer_roctx_LIBRARY)
|
|
|
|
#------------------------------------------------------------------------------#
|
|
|
|
if(roctracer_FOUND)
|
|
add_library(roctracer::roctracer INTERFACE IMPORTED)
|
|
add_library(roctracer::roctx INTERFACE IMPORTED)
|
|
set(roctracer_INCLUDE_DIRS ${roctracer_INCLUDE_DIR} ${roctracer_hsa_INCLUDE_DIR})
|
|
set(roctracer_LIBRARIES ${roctracer_LIBRARY} ${roctracer_roctx_LIBRARY})
|
|
set(roctracer_LIBRARY_DIRS ${roctracer_LIBRARY_DIR})
|
|
|
|
target_include_directories(roctracer::roctracer INTERFACE ${roctracer_INCLUDE_DIR} ${roctracer_hsa_INCLUDE_DIR})
|
|
target_include_directories(roctracer::roctx INTERFACE ${roctracer_INCLUDE_DIR} ${roctracer_hsa_INCLUDE_DIR})
|
|
|
|
target_link_libraries(roctracer::roctracer INTERFACE ${roctracer_LIBRARY})
|
|
target_link_libraries(roctracer::roctx INTERFACE ${roctracer_roctx_LIBRARY})
|
|
|
|
if(roctracer_kfdwrapper_LIBRARY)
|
|
list(APPEND roctracer_LIBRARIES ${roctracer_kfdwrapper_LIBRARY})
|
|
target_compile_definitions(roctracer::roctracer INTERFACE HOSTTRACE_ROCTRACER_LIBKFDWRAPPER=\"${roctracer_kfdwrapper_LIBRARY}\")
|
|
target_link_libraries(roctracer::roctracer INTERFACE ${roctracer_kfdwrapper_LIBRARY})
|
|
target_link_libraries(roctracer::roctx INTERFACE ${roctracer_kfdwrapper_LIBRARY})
|
|
endif()
|
|
|
|
if(roctracer_hsakmt_LIBRARY)
|
|
list(APPEND roctracer_LIBRARIES ${roctracer_hsakmt_LIBRARY})
|
|
target_link_libraries(roctracer::roctracer INTERFACE ${roctracer_hsakmt_LIBRARY})
|
|
target_link_libraries(roctracer::roctx INTERFACE ${roctracer_hsakmt_LIBRARY})
|
|
endif()
|
|
|
|
endif()
|
|
|
|
#------------------------------------------------------------------------------#
|
|
|
|
unset(_ROCM_PATHS)
|
|
|
|
#------------------------------------------------------------------------------#
|