GetCoreNum
函数功能
获取当前硬件平台的核数。若AI Core的架构为Cube、Vector分离架构,返回AI Core上的Vector核数;非分离架构返回AI Core的核数。
函数原型
1 | uint32_t GetCoreNum(void) const |
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
- |
- |
- |
返回值说明
针对
针对
约束说明
无
调用示例
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; } |
父主题: PlatformAscendC