Galantsev, Dmitrii
213ccc7e72
Signed-off-by: Galantsev, Dmitrii <dmitrii.galantsev@amd.com>
[ROCm/rdc commit: 3f2f92a37a]
71 строка
1.8 KiB
CMake
71 строка
1.8 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(hiprand 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
|
|
hip::hiprand
|
|
hsa-runtime64::hsa-runtime64
|
|
amd_smi
|
|
)
|
|
endif()
|