David Galiffi
80 lines
2.6 KiB
CMake
80 lines
2.6 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/rocm_smi)
|
|
if(EXISTS ${_DIR})
|
|
get_filename_component(_ABS_DIR "${_DIR}" REALPATH)
|
|
list(APPEND _ROCM_SMI_PATHS ${_ABS_DIR})
|
|
endif()
|
|
endforeach()
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
find_path(
|
|
rocm-smi_ROOT_DIR
|
|
NAMES include/rocm_smi/rocm_smi.h
|
|
HINTS ${_ROCM_SMI_PATHS}
|
|
PATHS ${_ROCM_SMI_PATHS}
|
|
PATH_SUFFIXES rocm_smi)
|
|
|
|
mark_as_advanced(rocm-smi_ROOT_DIR)
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
find_path(
|
|
rocm-smi_INCLUDE_DIR
|
|
NAMES rocm_smi/rocm_smi.h
|
|
HINTS ${rocm-smi_ROOT_DIR} ${_ROCM_SMI_PATHS}
|
|
PATHS ${rocm-smi_ROOT_DIR} ${_ROCM_SMI_PATHS}
|
|
PATH_SUFFIXES include rocm_smi/include)
|
|
|
|
mark_as_advanced(rocm-smi_INCLUDE_DIR)
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
find_library(
|
|
rocm-smi_LIBRARY
|
|
NAMES rocm_smi64 rocm_smi
|
|
HINTS ${rocm-smi_ROOT_DIR} ${_ROCM_SMI_PATHS}
|
|
PATHS ${rocm-smi_ROOT_DIR} ${_ROCM_SMI_PATHS}
|
|
PATH_SUFFIXES rocm_smi/lib rocm_smi/lib64 lib lib64)
|
|
|
|
if(rocm-smi_LIBRARY)
|
|
get_filename_component(rocm-smi_LIBRARY_DIR "${rocm-smi_LIBRARY}" PATH CACHE)
|
|
endif()
|
|
|
|
mark_as_advanced(rocm-smi_LIBRARY)
|
|
|
|
# ----------------------------------------------------------------------------------------#
|
|
|
|
find_package_handle_standard_args(rocm-smi DEFAULT_MSG rocm-smi_ROOT_DIR
|
|
rocm-smi_INCLUDE_DIR rocm-smi_LIBRARY)
|
|
|
|
# ------------------------------------------------------------------------------#
|
|
|
|
if(rocm-smi_FOUND)
|
|
add_library(rocm-smi::rocm-smi INTERFACE IMPORTED)
|
|
add_library(rocm-smi::roctx INTERFACE IMPORTED)
|
|
set(rocm-smi_INCLUDE_DIRS ${rocm-smi_INCLUDE_DIR})
|
|
set(rocm-smi_LIBRARIES ${rocm-smi_LIBRARY})
|
|
set(rocm-smi_LIBRARY_DIRS ${rocm-smi_LIBRARY_DIR})
|
|
|
|
target_include_directories(rocm-smi::rocm-smi INTERFACE ${rocm-smi_INCLUDE_DIR})
|
|
target_link_libraries(rocm-smi::rocm-smi INTERFACE ${rocm-smi_LIBRARY})
|
|
|
|
endif()
|
|
|
|
# ------------------------------------------------------------------------------#
|
|
|
|
unset(_ROCM_SMI_PATHS)
|
|
|
|
# ------------------------------------------------------------------------------#
|