rocm-systems/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()