* 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.
*/
#include <cstdint>
#include <string>
#if defined(__NPU_ARCH__) && (__NPU_ARCH__ == 3002)
static const char* g_core_num_aic = "1";
static const char* g_core_num_aiv = "1";
static const char* g_core_num_cub = "1";
static const char* g_core_type_list = "AICore,CubeCore,VectorCore";
static const char* g_chip_version = "Ascend310B";
static const char* g_npu_arch = "3002";
static uint32_t g_core_num = 1;
static const char* g_vector_len = "0";
#elif defined(__NPU_ARCH__) && (__NPU_ARCH__ == 2201)
static const char* g_core_num_aic = "24";
static const char* g_core_num_aiv = "48";
static const char* g_core_num_cub = "24";
static const char* g_core_type_list = "CubeCore,VectorCore";
static const char* g_chip_version = "Ascend910B";
static const char* g_npu_arch = "2201";
static uint32_t g_core_num = 48;
static const char* g_vector_len = "0";
#elif defined(__NPU_ARCH__) && (__NPU_ARCH__ == 2002)
static const char* g_core_num_aic = "10";
static const char* g_core_num_aiv = "8";
static const char* g_core_num_cub = "10";
static const char* g_core_type_list = "AICore,VectorCore";
static const char* g_chip_version = "Ascend310P";
static const char* g_npu_arch = "2202";
static uint32_t g_core_num = 10;
static const char* g_vector_len = "0";
#elif defined(__NPU_ARCH__) && (__NPU_ARCH__ == 3510 || __NPU_ARCH__ == 5102)
static const char* g_core_num_aic = "24";
static const char* g_core_num_aiv = "48";
static const char* g_core_num_cub = "24";
static const char* g_core_type_list = "AICore,VectorCore";
static const char* g_chip_version = "Ascend950";
static const char* g_npu_arch = "3510";
static uint32_t g_core_num = 48;
static const char* g_vector_len = "256";
#else
static const char* g_core_num_aic = "32";
static const char* g_core_num_aiv = "0";
static const char* g_core_num_cub = "32";
static const char* g_core_type_list = "AICore";
static const char* g_chip_version = "Ascend910";
static const char* g_npu_arch = "1001";
static uint32_t g_core_num = 32;
static const char* g_vector_len = "0";
#endif
void platfrom_stub_set_num_aic(const char* num) { g_core_num_aic = num; }
void platfrom_stub_set_num_aiv(const char* num) { g_core_num_aiv = num; }
void platfrom_stub_set_num_cub(const char* num) { g_core_num_cub = num; }
void platfrom_stub_set_ctl(const char* num) { g_core_type_list = num; }
void platfrom_stub_set_npuarch(const char* num) { g_npu_arch = num; }
void platfrom_stub_set_chip_version(const char* num) { g_chip_version = num; }
void platfrom_stub_set_num(uint32_t num) { g_core_num = num; }
namespace fe {
enum class LocalMemType { L0_A = 0, L0_B = 1, L0_C = 2, L1 = 3, L2 = 4, UB = 5, HBM = 6, RESERVED };
class OptionalInfos {
public:
bool Init();
};
class PlatFormInfos {
public:
uint32_t GetCoreNum(void) const;
uint32_t GetCoreNumAiv(void) const;
bool GetPlatformRes(const std::string& label, const std::string& key, std::string& val);
bool GetPlatformResWithLock(const std::string& label, const std::string& key, std::string& val);
void GetLocalMemSize(const LocalMemType& mem_type, uint64_t& size);
void GetLocalMemBw(const LocalMemType& mem_type, uint64_t& bw_size);
void SetCoreNumByCoreType(const std::string& core_type);
};
void PlatFormInfos::SetCoreNumByCoreType(const std::string& core_type) { return; }
uint32_t PlatFormInfos::GetCoreNum(void) const { return g_core_num; }
uint32_t PlatFormInfos::GetCoreNumAiv(void) const { return g_core_num; }
bool PlatFormInfos::GetPlatformRes(const std::string& label, const std::string& key, std::string& val)
{
if (label.compare("SoCInfo") != 0 && label.compare("version") != 0 && label.compare("AICoreSpec") != 0) {
val = "0";
return false;
}
if (key.compare("core_type_list") == 0) {
val = g_core_type_list;
} else if (key.compare("ai_core_cnt") == 0) {
val = g_core_num_aic;
} else if (key.compare("cube_core_cnt") == 0) {
val = g_core_num_cub;
} else if (key.compare("vector_core_cnt") == 0) {
val = g_core_num_aiv;
} else if (key.compare("Short_SoC_version") == 0) {
val = g_chip_version;
} else if (key.compare("NpuArch") == 0) {
val = g_npu_arch;
} else if (key.compare("vector_reg_width") == 0) {
val = g_vector_len;
} else {
val = "0";
return false;
}
return true;
}
bool PlatFormInfos::GetPlatformResWithLock(const std::string& label, const std::string& key, std::string& val)
{
return GetPlatformRes(label, key, val);
}
#if defined(__NPU_ARCH__) && (__NPU_ARCH__ == 3002)
void PlatFormInfos::GetLocalMemSize(const LocalMemType& mem_type, uint64_t& size)
{
switch (mem_type) {
case LocalMemType::L0_A:
size = 65536;
break;
case LocalMemType::L0_B:
size = 65536;
break;
case LocalMemType::L0_C:
size = 131072;
break;
case LocalMemType::UB:
size = 253952;
break;
case LocalMemType::L1:
size = 1048576;
break;
case LocalMemType::L2:
size = 4194304;
break;
case LocalMemType::HBM:
size = 0;
break;
default:
size = 0;
break;
}
}
void PlatFormInfos::GetLocalMemBw(const LocalMemType& mem_type, uint64_t& bw_size)
{
switch (mem_type) {
case LocalMemType::L2:
bw_size = 256;
break;
case LocalMemType::HBM:
bw_size = 17;
break;
default:
bw_size = 0;
break;
}
}
#elif defined(__NPU_ARCH__) && (__NPU_ARCH__ == 2201)
void PlatFormInfos::GetLocalMemSize(const LocalMemType& mem_type, uint64_t& size)
{
switch (mem_type) {
case LocalMemType::L0_A:
size = 65536;
break;
case LocalMemType::L0_B:
size = 65536;
break;
case LocalMemType::L0_C:
size = 131072;
break;
case LocalMemType::UB:
size = 196608;
break;
case LocalMemType::L1:
size = 524288;
break;
case LocalMemType::L2:
size = 201326592;
break;
case LocalMemType::HBM:
size = 0;
break;
default:
size = 0;
break;
}
}
void PlatFormInfos::GetLocalMemBw(const LocalMemType& mem_type, uint64_t& bw_size)
{
switch (mem_type) {
case LocalMemType::L2:
bw_size = 110;
break;
case LocalMemType::HBM:
bw_size = 32;
break;
default:
bw_size = 0;
break;
}
}
#elif defined(__NPU_ARCH__) && (__NPU_ARCH__ == 2002)
void PlatFormInfos::GetLocalMemSize(const LocalMemType& mem_type, uint64_t& size)
{
switch (mem_type) {
case LocalMemType::L0_A:
size = 65536;
break;
case LocalMemType::L0_B:
size = 65536;
break;
case LocalMemType::L0_C:
size = 262144;
break;
case LocalMemType::UB:
size = 262144;
break;
case LocalMemType::L1:
size = 1048576;
break;
case LocalMemType::L2:
size = 16777216;
break;
case LocalMemType::HBM:
size = 0;
break;
default:
size = 0;
break;
}
}
void PlatFormInfos::GetLocalMemBw(const LocalMemType& mem_type, uint64_t& bw_size)
{
switch (mem_type) {
case LocalMemType::L2:
bw_size = 114;
break;
case LocalMemType::HBM:
bw_size = 17;
break;
default:
bw_size = 0;
break;
}
}
#else
void PlatFormInfos::GetLocalMemSize(const LocalMemType& mem_type, uint64_t& size)
{
switch (mem_type) {
case LocalMemType::L0_A:
size = 65536;
break;
case LocalMemType::L0_B:
size = 65536;
break;
case LocalMemType::L0_C:
size = 262144;
break;
case LocalMemType::UB:
size = 262144;
break;
case LocalMemType::L1:
size = 1048576;
break;
case LocalMemType::L2:
size = 33554432;
break;
case LocalMemType::HBM:
size = 0;
break;
default:
size = 0;
break;
}
}
void PlatFormInfos::GetLocalMemBw(const LocalMemType& mem_type, uint64_t& bw_size)
{
switch (mem_type) {
case LocalMemType::L2:
bw_size = 110;
break;
case LocalMemType::HBM:
bw_size = 32;
break;
default:
bw_size = 0;
break;
}
}
#endif
class PlatformInfoManager {
public:
static PlatformInfoManager& GeInstance();
static PlatformInfoManager& Instance();
uint32_t InitializePlatformInfo();
uint32_t GetPlatformInfos(
const std::string SoCVersion, PlatFormInfos& platform_info, OptionalInfos& opti_compilation_info);
uint32_t InitRuntimePlatformInfos(const std::string& SoCVersion);
uint32_t GetRuntimePlatformInfosByDevice(const uint32_t& deviceId, PlatFormInfos& platform_info, bool flag = false);
};
uint32_t PlatformInfoManager::InitRuntimePlatformInfos(const std::string& soCVersion) { return 0; }
uint32_t PlatformInfoManager::GetRuntimePlatformInfosByDevice(
const uint32_t& deviceId, PlatFormInfos& platform_info, bool flag)
{
return 0;
}
uint32_t PlatformInfoManager::GetPlatformInfos(
const std::string SoCVersion, PlatFormInfos& platform_info, OptionalInfos& opti_compilation_info)
{
return 0;
}
PlatformInfoManager& PlatformInfoManager::GeInstance()
{
static PlatformInfoManager pf;
return pf;
}
PlatformInfoManager& PlatformInfoManager::Instance()
{
static PlatformInfoManager pf;
return pf;
}
}
