GetCoreNum

函数功能

获取当前硬件平台的核数。若AI Core的架构为Cube、Vector分离架构，返回AI Core上的Vector核数；非分离架构返回AI Core的核数。

函数原型

1

uint32_t GetCoreNum(void) const

参数说明

无

返回值说明

针对Atlas 训练系列产品，非Cube、Vector分离架构，返回AI Core的核数

针对Atlas 推理系列产品，非Cube、Vector分离架构，返回AI Core的核数

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件，Cube、Vector分离架构，返回AI Core上的Vector核数

约束说明

无

调用示例

1
2
3
4
5
6
7

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