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;
}
父主题: 平台信息获取