rocm-systems/projects/rdc/cmake_modules/Findrvs.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()