# Copyright (c) Advanced Micro Devices, Inc.
# SPDX-License-Identifier: MIT

cmake_minimum_required(VERSION 3.25)
project(hipdnn_install_tests LANGUAGES CXX)
set(CMAKE_CXX_STANDARD 17)

enable_testing()

# Add a CMake install test for a given hipDNN package and header.
function(add_hipdnn_install_test)
    cmake_parse_arguments(ARG "" "PACKAGE;HEADER" "" ${ARGN})

    find_package(${ARG_PACKAGE} CONFIG REQUIRED)
    message(STATUS "${ARG_PACKAGE} found")

    set(test_name "test_${ARG_PACKAGE}")
    set(header "${ARG_HEADER}")
    set(pkg_name "${ARG_PACKAGE}")
    configure_file(
        "${CMAKE_CURRENT_SOURCE_DIR}/test_template.cpp.in"
        "${CMAKE_CURRENT_BINARY_DIR}/${test_name}.cpp"
        @ONLY
    )
    add_executable(${test_name} "${CMAKE_CURRENT_BINARY_DIR}/${test_name}.cpp")
    target_link_libraries(${test_name} PRIVATE ${ARG_PACKAGE})
    add_test(NAME ${test_name} COMMAND ${test_name})
endfunction()

add_hipdnn_install_test(PACKAGE hipdnn_backend        HEADER hipdnn_backend.h)
add_hipdnn_install_test(PACKAGE hipdnn_data_sdk        HEADER hipdnn_data_sdk/types.hpp)
add_hipdnn_install_test(PACKAGE hipdnn_flatbuffers_sdk HEADER hipdnn_flatbuffers_sdk/data_objects/graph_generated.h)
add_hipdnn_install_test(PACKAGE hipdnn_frontend   HEADER hipdnn_frontend.hpp)
add_hipdnn_install_test(PACKAGE hipdnn_plugin_sdk HEADER hipdnn_plugin_sdk/PluginApi.h)
add_hipdnn_install_test(PACKAGE hipdnn_test_sdk   HEADER hipdnn_test_sdk/utilities/Seeds.hpp)