# -----------------------------------------------------------------------------------------------------------
# Copyright (c) 2026 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.
# -----------------------------------------------------------------------------------------------------------

set(_MLA_SOURCES
    ${CMAKE_CURRENT_SOURCE_DIR}/mla.cpp
    ${CMAKE_CURRENT_SOURCE_DIR}/tiling/mla_tiling.cpp
)
set_source_files_properties(${_MLA_SOURCES} PROPERTIES LANGUAGE ASC)

add_kernel(NAME mla
    NPU_ARCH_LIST 2201
    KERNEL_TYPE prebuilt
    ${_MLA_SOURCES}
)

foreach(_SRC ${_MLA_SOURCES})
    get_filename_component(_SRC_NAME ${_SRC} NAME)
    foreach(_ARCH 2201)
        foreach(_MS_SUFFIX "" "_ms")
            set(_LIB_NAME "catlass_kernel_${_ARCH}_mla${_MS_SUFFIX}")
            if(TARGET ${_LIB_NAME})
                target_compile_options(${_LIB_NAME} PRIVATE
                    "SHELL:$<$<COMPILE_LANGUAGE:ASC>:-mllvm -cce-aicore-record-overflow=true>"
                    "SHELL:$<$<COMPILE_LANGUAGE:ASC>:-mllvm -cce-aicore-dcci-insert-for-scalar=false>"
                )
                target_include_directories(${_LIB_NAME} PRIVATE
                    ${CMAKE_CURRENT_SOURCE_DIR}
                    ${CMAKE_CURRENT_SOURCE_DIR}/tiling
                    ${CMAKE_CURRENT_SOURCE_DIR}/../../include
                )
            endif()
        endforeach()
    endforeach()
endforeach()