GetCoreNumAiv

函数功能

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

函数原型

uint32_t GetCoreNumAiv(void) const;

参数说明

参数

输入/输出

说明

-

-

-

返回值说明

返回Vector Core的核数。

约束说明

调用示例

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