GetCurNpuArch

功能说明

获取当前硬件平台芯片架构版本号。

函数原型

NpuArch GetCurNpuArch(void) const

参数说明

无

返回值

当前硬件平台架构号的枚举类。该枚举类和AI处理器型号的对应关系请通过CANN软件安装后文件存储路径下include/platform/soc_spec.h头文件获取。

产品	NPU_ARCH
Atlas A3 训练系列产品/Atlas A3 推理系列产品	DAV_2201
Atlas A2 训练系列产品/Atlas A2 推理系列产品	DAV_2201
Atlas 200I/500 A2 推理产品	DAV_3002
Atlas 推理系列产品AI Core	DAV_2002
Atlas 推理系列产品Vector Core	DAV_2002
Atlas 训练系列产品	DAV_1001

约束说明

无

调用示例

ge::graphStatus TilingXXX(gert::TilingContext* context) {
    auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
    auto npuArch = ascendcPlatform.GetCurNpuArch();
    // 根据所获得的版本型号自行设计Tiling策略
    // DAV_XXX请替换为实际的架构号
    if (socVersion == NpuArch::DAV_XXXX) {
        // ...
    }
    return ret;
}

父主题： PlatformAscendC