GetCoreNumAiv

Function

Obtains the number of vector cores on AI Cores of the current hardware platform. If the AI Core architecture is cube/vector separated, the number of vector cores is returned. If the architecture is coupled, the number of AI Cores is returned.

Prototype

1

uint32_t GetCoreNumAiv(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 separated mode, the number of Vector Cores is returned.

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

Restrictions

None

Examples

1
2
3
4
5
6
7
8

ge::graphStatus TilingXXX(gert::TilingContext* context) {
    auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo());
    auto aicNum = ascendcPlatform.GetCoreNumAic();
    auto aivNum = ascendcPlatform.GetCoreNumAiv();
    // ... Split by aivNum.
    context->SetBlockDim(ascendcPlatform.CalcTschBlockDim(aivNum, aicNum, aivNum));
    return ret;
}