# ----------------------------------------------------------------------------
# Copyright (c) 2025 Huawei Technologies Co., Ltd.
# This program is free software, you can redistribute it and/or modify it under the terms and conditions of
# CANN Open Software License Agreement Version 2.0 (the "License").
# Please refer to the License for details. You may not use this file except in compliance with the License.
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
# See LICENSE in the root of the software repository for the full text of the License.
# ----------------------------------------------------------------------------

cmake_minimum_required(VERSION 3.5.1)
project(llm_datadist_sample)
set(CMAKE_SKIP_RPATH TRUE)
set(HIXL_INC_DIR "${HIXL_CODE_DIR}/include")
add_definitions(-D_GLIBCXX_USE_CXX11_ABI=0)

# 只定义目标名称列表（与cpp文件名前缀相同）
set(targets_list
    "prompt_pull_cache_and_blocks"
    "decoder_pull_cache_and_blocks"
    "prompt_push_cache_and_blocks"
    "decoder_push_cache_and_blocks"
    "prompt_switch_roles"
    "decoder_switch_roles"
    "hixl_example_d2rd_multiproc"
    "fabric_mem_d2d"
    "hixl_example_d2rh"
    "hixl_example_d2rd"
)

foreach(target_name IN LISTS targets_list)
    # 自动构建源文件路径（目标名.cpp）
    set(source_file "${target_name}.cpp")

    if(NOT TARGET ${target_name})
        add_executable(${target_name} ${source_file})
        target_include_directories(${target_name} PRIVATE
            ${HIXL_INC_DIR}
            ${ASCEND_INSTALL_PATH}/include
        )
        target_compile_options(${target_name} PRIVATE
            -ftrapv
            -O2
            -fno-common
            -Wfloat-equal
            -Wall
            -Werror
            -Wextra
        )
        if(${target_name} MATCHES "hixl_example_d2rd_multiproc|fabric_mem_d2d|hixl_example_d2rh|hixl_example_d2rd")
            target_link_libraries(${target_name} PRIVATE
                cann_hixl
                acl_rt_headers
                acl_rt
                -lpthread
                -lrt
            )
        else()
            target_link_libraries(${target_name} PRIVATE
                llm_datadist
                acl_rt_headers
                acl_rt
                -lpthread
                -lrt
            )
        endif()
        target_link_libraries(${target_name} PRIVATE ascend_hal)
    endif()
endforeach()