/**
* 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.
*/

/*!
 * \file kernel_operator_sys_var_intf_impl.h
 * \brief
 */
#if !defined(__ASCENDC_INCLUDE_INTERNAL_HEADERS__)
#pragma message("impl/basic_api/kernel_operator_sys_var_intf_impl.h is an internal header file and must not be used directly. Functions or variables defined in this file may be removed in the future. Please use \"#include \"basic_api/kernel_operator_sys_var_intf.h\"\" and use public functions or variables defined in interface headers files.")
#define __ASCENDC_INCLUDE_INTERNAL_HEADERS__
#define __UNDEF_ASCENDC_INCLUDE_INTERNAL_HEADERS_KERNEL_OPERATOR_SYS_VAR_INTF_IMPL_H__
#endif
#ifndef ASCENDC_MODULE_OPERATOR_SYS_VAR_INTERFACE_IMPL_H
#define ASCENDC_MODULE_OPERATOR_SYS_VAR_INTERFACE_IMPL_H

#include <cstdint>
#include "kernel_macros.h"

#if __NPU_ARCH__ == 1001
#include "dav_c100/kernel_operator_sys_var_impl.h"
#elif __NPU_ARCH__ == 2002
#include "dav_m200/kernel_operator_sys_var_impl.h"
#elif __NPU_ARCH__ == 2201
#include "dav_c220/kernel_operator_sys_var_impl.h"
#elif __NPU_ARCH__ == 3002
#include "dav_m300/kernel_operator_sys_var_impl.h"
#elif __NPU_ARCH__ == 3102
#include "dav_m310/kernel_operator_sys_var_impl.h"
#elif __NPU_ARCH__ == 3510
#include "dav_3510/kernel_operator_sys_var_impl.h"
#elif (__NPU_ARCH__ == 5102)
#include "dav_m510/kernel_operator_sys_var_impl.h"
#elif __NPU_ARCH__ == 3003
#include "dav_l300/kernel_operator_sys_var_impl.h"
#elif __NPU_ARCH__ == 3113
#include "dav_l311/kernel_operator_sys_var_impl.h"
#endif

#if __NPU_ARCH__ == 2201 || (__NPU_ARCH__ == 3510) || (__NPU_ARCH__ == 5102)
#ifdef __SUPER_KERNEL_DYNAMIC_BLOCK_NUM__
__BLOCK_LOCAL__ __inline__ uint32_t g_super_kernel_dynamic_block_num;
#endif
#endif

namespace sk {
// transmit block local variable information in SK "sub-kernel"
struct SkSystemArgs {
    uint16_t skBlockIdx;    // sub-kernel block idx
    uint16_t skNumBlocks;    // sub-kernel block num
    uint16_t skTaskSyncCfg; // sync type for "early-start"
    uint8_t reserve[10];
    inline __aicore__ uint16_t SkGetBlockIdx() {
        return skBlockIdx;
    }
    inline __aicore__ uint16_t SkGetNumBlocks() {
        return skNumBlocks;
    }
    inline __aicore__ uint16_t SkGetTaskSyncCfg() {
        return skTaskSyncCfg;
    }
    inline __aicore__ void SkSetTaskSyncCfg(uint16_t config) {
        skTaskSyncCfg = config;
    }
};
}

namespace AscendC {

__aicore__ inline int64_t GetBlockIdx()
{
    return GetBlockIdxImpl();
}

__aicore__ inline int64_t GetBlockNum()
{
#ifdef __SUPER_KERNEL_STATIC_BLOCK_NUM__
    return __SUPER_KERNEL_STATIC_BLOCK_NUM__;
#elif defined(__SUPER_KERNEL_DYNAMIC_BLOCK_NUM__)
    return g_super_kernel_dynamic_block_num;
#else
    return get_block_num();
#endif
}

__aicore__ inline int64_t GetSubBlockIdx()
{
    return GetSubBlockIdxImpl();
}

__aicore__ inline int64_t GetTaskRatio()
{
    return GetTaskRationImpl();
}

__aicore__ inline int64_t GetTaskRation()
{
    return GetTaskRatio();
}

__aicore__ inline void GetArchVersion(uint32_t& coreVersion)
{
    GetArchVersionImpl(coreVersion);
}

__aicore__ inline int64_t GetSubBlockNum()
{
    return GetSubBlockNumImpl();
}

__aicore__ inline int64_t GetProgramCounter()
{
    return GetProgramCounterImpl();
}

__aicore__ inline void Trap()
{
    TrapImpl();
}

__aicore__ inline int64_t GetSystemCycle()
{
    return GetSystemCycleImpl();
}

#if (__NPU_ARCH__ == 3510) || (__NPU_ARCH__ == 5102) || (__NPU_ARCH__ == 3003) || (__NPU_ARCH__ == 3113)
template <SpecialPurposeReg spr>
__aicore__ inline int64_t GetSpr(){
    return GetSprImpl<spr>();
}

template <SpecialPurposeReg spr>
__aicore__ inline void ClearSpr(){
    ClearSprImpl<spr>();
}
#endif
}  // namespace AscendC
#endif  // ASCENDC_MODULE_OPERATOR_SYS_VAR_INTERFACE_IMPL_H
#if defined(__UNDEF_ASCENDC_INCLUDE_INTERNAL_HEADERS_KERNEL_OPERATOR_SYS_VAR_INTF_IMPL_H__)
#undef __ASCENDC_INCLUDE_INTERNAL_HEADERS__
#undef __UNDEF_ASCENDC_INCLUDE_INTERNAL_HEADERS_KERNEL_OPERATOR_SYS_VAR_INTF_IMPL_H__
#endif