Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 11 additions & 3 deletions cmake_modules/Findrvs.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,16 @@ if(NOT DEFINED ROCM_DIR)
endif()
list(APPEND CMAKE_PREFIX_PATH ${ROCM_DIR})

find_package(${NAME} NAMES ${NAME} rocm_validation_suite REQUIRED CONFIG)

find_library(
${NAME}_LIBRARY
NAMES
${ROCM_VALIDATION_SUITE_LIBRARY} # this is set by the RVS package
${NAME}
${NAME}64
${NAME}lib # RVS is special and is named librvslib.so
HINTS "${ROCM_DIR}" "${ROCM_VALIDATION_SUITE_LIB_DIR}"
REQUIRED
REGISTRY_VIEW
BOTH
Expand All @@ -24,8 +28,9 @@ if(NOT DEFINED (${NAME}_INCLUDE_DIR))
find_path(
${NAME}_INCLUDE_DIR
NAMES ${NAME}.h
HINTS "${ROCM_DIR}/include"
HINTS "${ROCM_DIR}/include" "${ROCM_VALIDATION_SUITE_INCLUDE_DIR}"
PATH_SUFFIXES ${NAME} ${NAME}/inc
REQUIRED
)
endif()

Expand All @@ -36,6 +41,7 @@ find_package_handle_standard_args(
REQUIRED_VARS ${NAME}_LIBRARY ${NAME}_INCLUDE_DIR
)

# rvs::rvs is not defined, but found rvs package, so we define our own library
if(${NAME}_FOUND AND NOT TARGET ${NAME}::${NAME})
add_library(${NAME}::${NAME} UNKNOWN IMPORTED)
set_target_properties(
Expand All @@ -45,23 +51,25 @@ if(${NAME}_FOUND AND NOT TARGET ${NAME}::${NAME})
INTERFACE_COMPILE_OPTIONS "${PC_${NAME}_CFLAGS_OTHER}"
INTERFACE_INCLUDE_DIRECTORIES "${${NAME}_INCLUDE_DIR}"
)
find_library(rocm-core NAMES rocm-core REQUIRED)
find_package(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}
rocm-core
yaml-cpp
roc::rocblas
roc::hipblaslt
hsakmt::hsakmt
hip::amdhip64
hip::hiprand
hsa-runtime64::hsa-runtime64
amd_smi
)
Expand Down
Loading