Galantsev, Dmitrii
39687e8d96
Change-Id: I65095cc3d04fc2a5daeee5c809f635cb1662822f Signed-off-by: Galantsev, Dmitrii <dmitrii.galantsev@amd.com> Revert "Disable RVS as the error scares people" This reverts commitf3450f61bf. Change-Id: I5086c25772444aa3bfc4c10abc1ea58d3f3f1f27 Signed-off-by: Galantsev, Dmitrii <dmitrii.galantsev@amd.com> [ROCm/rdc commit:dd50027748]
52 baris
1.6 KiB
CMake
52 baris
1.6 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()
|