GetCurNpuArch

功能说明

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

函数原型

Text

NpuArch GetCurNpuArch(void) const

参数说明

无

返回值

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

• Ascend 950PR/Ascend 950DT：DAV_3510

• 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

约束说明

无

调用示例

Text

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;
}