GetCoreNum

函数功能

获取当前硬件平台的核数。

函数原型

uint32_t GetCoreNum(void) const;

参数说明

返回值说明

AI Core的数量。

针对Atlas 推理系列产品、Atlas 训练系列产品，AI Core指1个Cube Core和1个Vector Core。

针对Atlas A2训练系列产品，AI Core指1个Cube Core和2个Vector Core。

约束说明

无

调用示例

ge::graphStatus TilingXXX(gert::TilingContext* context) {
    auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
    auto coreNum = ascendcPlatform.GetCoreNum();
    // ... 根据核数自行设计Tiling策略
    context->SetBlockDim(coreNum);
    return ret;
}