GetCoreNum

Function

Obtains the number of cores on the current hardware platform. If the AI Core architecture is in the decoupled mode of Cube and Vector, the number of Vector Cores is returned. In the coupled mode, the number of AI Cores is returned.

Prototype

1

uint32_t GetCoreNum(void) const

Parameters

None

Returns

Atlas training products: In coupled mode, the number of AI Cores is returned.

Atlas inference products: In coupled mode, the number of AI Cores is returned.

Atlas A2 training products/Atlas A2 inference products: In decoupled mode, the number of Vector Cores is returned.

Atlas A3 training products/Atlas A3 inference products: In decoupled mode, the number of Vector Cores is returned.

Restrictions

None

Examples

1
2
3
4
5
6
7

ge::graphStatus TilingXXX(gert::TilingContext* context) {
    auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
    auto coreNum = ascendcPlatform.GetCoreNum();
    // ... Design the tiling policy based on the number of cores.
    context->SetBlockDim(coreNum);
    return ret;
}