#
# Copyright © Advanced Micro Devices, Inc., or its affiliates.
#
# Redistribution and use in source and binary forms, with or without modification,
# are permitted provided that the following conditions are met:
# 1. Redistributions of source code must retain the above copyright notice, this
#    list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright notice,
#    this list of conditions and the following disclaimer in the documentation
#    and/or other materials provided with the distribution.
# 3. Neither the name of the copyright holder nor the names of its contributors
#    may be used to endorse or promote products derived from this software without
#    specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (
# INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
# OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
# EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#

# Create bench config
create_bench_config(${CMAKE_CURRENT_SOURCE_DIR}/configs ${CMAKE_CURRENT_SOURCE_DIR}/include/bench_config.hh.in ${CMAKE_CURRENT_BINARY_DIR}/bench_config.hh)

# Include dlp and utils headers
set(BENCH_INCLUDE_DIRS
    ${PROJECT_SOURCE_DIR}/tests/include
    ${PROJECT_BINARY_DIR}/tests/classic
    ${CMAKE_CURRENT_BINARY_DIR}
    ${CMAKE_CURRENT_SOURCE_DIR}/include
    ${CLASSIC_INCLUDES}
)

# Fetch google benchmark
fetch_benchmark()

# Fetch yaml-cpp for test configuration
fetch_yaml_cpp()

# Collect framework source files
dlp_glob_sources_recursive(FRAMEWORK_SOURCES "${PROJECT_SOURCE_DIR}/tests/framework")

# Collect utils source files
dlp_glob_sources_recursive(UTILS_SOURCES "${PROJECT_SOURCE_DIR}/tests/utils")

# Collect ref source files
dlp_glob_sources_recursive(ADAPTORS_SOURCES "${PROJECT_SOURCE_DIR}/tests/adaptors")

# Collect bench infrastructure source files
set(BENCH_INFRA_SOURCES
    ${CMAKE_CURRENT_SOURCE_DIR}/src/bench_types.cc
    ${CMAKE_CURRENT_SOURCE_DIR}/src/bench_metrics.cc
)

# Collect benchmark source files
dlp_glob_sources(OUTPUT BENCHMARK_SOURCES DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}")

# Build all benchmarks
foreach(benchmark_source ${BENCHMARK_SOURCES})
    get_filename_component(benchmark_name ${benchmark_source} NAME_WE)

    dlp_add_benchmark(
        NAME ${benchmark_name}
        SOURCES ${benchmark_source} ${BENCH_INFRA_SOURCES} ${FRAMEWORK_SOURCES} ${UTILS_SOURCES} ${ADAPTORS_SOURCES}
        DEPENDS yaml-cpp dlp::openmp
    )
    dlp_set_global_compile_flags(${benchmark_name} VISIBILITY PUBLIC)

    # Link NUMA library for bench_gemm (optimized implementation)
    if(${benchmark_name} STREQUAL "bench_gemm")
        target_link_libraries(${benchmark_name} PRIVATE numa)
    endif()

    # Disable static linking for benchmarks
    set_target_properties(${benchmark_name} PROPERTIES
        LINK_SEARCH_START_STATIC FALSE
        LINK_SEARCH_END_STATIC FALSE
    )
endforeach()
