Galantsev, Dmitrii
fa8b89f4ae
Change-Id: I08e71fc5060b1f6e0168225cc5fe66886c2044bd Signed-off-by: Galantsev, Dmitrii <dmitrii.galantsev@amd.com>
57 rindas
1.7 KiB
CMake
57 rindas
1.7 KiB
CMake
# This module provides a rvs::rvs package
|
|
# You can specify the ROCM directory by setting ROCM_DIR
|
|
|
|
set(NAME rvs)
|
|
|
|
if(NOT DEFINED ROCM_DIR)
|
|
set(ROCM_DIR "/opt/rocm")
|
|
endif()
|
|
list(APPEND CMAKE_PREFIX_PATH ${ROCM_DIR})
|
|
|
|
find_library(
|
|
${NAME}_LIBRARY
|
|
NAMES ${NAME} ${NAME}64 ${NAME}lib # RVS is special and is named librvslib.so
|
|
REQUIRED REGISTRY_VIEW BOTH
|
|
PATH_SUFFIXES lib)
|
|
|
|
if(NOT DEFINED (${NAME}_INCLUDE_DIR))
|
|
find_path(
|
|
${NAME}_INCLUDE_DIR
|
|
NAMES ${NAME}.h
|
|
HINTS "${ROCM_DIR}/include"
|
|
PATH_SUFFIXES ${NAME} ${NAME}/inc)
|
|
endif()
|
|
|
|
include(FindPackageHandleStandardArgs)
|
|
find_package_handle_standard_args(
|
|
${NAME}
|
|
FOUND_VAR ${NAME}_FOUND
|
|
REQUIRED_VARS ${NAME}_LIBRARY ${NAME}_INCLUDE_DIR)
|
|
|
|
if(${NAME}_FOUND AND NOT TARGET ${NAME}::${NAME})
|
|
add_library(${NAME}::${NAME} UNKNOWN IMPORTED)
|
|
set_target_properties(
|
|
${NAME}::${NAME}
|
|
PROPERTIES IMPORTED_LOCATION "${${NAME}_LIBRARY}"
|
|
INTERFACE_COMPILE_OPTIONS "${PC_${NAME}_CFLAGS_OTHER}"
|
|
INTERFACE_INCLUDE_DIRECTORIES "${${NAME}_INCLUDE_DIR}")
|
|
find_library(rocm-core NAMES rocm-core REQUIRED)
|
|
find_package(yaml-cpp REQUIRED)
|
|
find_package(rocblas REQUIRED)
|
|
find_package(hipblaslt REQUIRED)
|
|
find_package(hsakmt REQUIRED)
|
|
find_package(hip REQUIRED)
|
|
find_package(hsa-runtime64 REQUIRED)
|
|
find_package(amd_smi REQUIRED)
|
|
target_link_libraries(
|
|
${NAME}::${NAME}
|
|
INTERFACE ${rocm-core}
|
|
yaml-cpp
|
|
roc::rocblas
|
|
roc::hipblaslt
|
|
hsakmt::hsakmt
|
|
hip::amdhip64
|
|
hsa-runtime64::hsa-runtime64
|
|
amd_smi)
|
|
endif()
|