GetCoreNumAiv
函数功能
获取当前硬件平台AI Core中Vector核数。若AI Core的架构为Cube、Vector分离架构,返回AI Core上的Vector核数;非分离架构返回AI Core的核数。
函数原型
uint32_t GetCoreNumAiv(void) const;
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
- |
- |
- |
返回值说明
针对Atlas 训练系列产品,非Cube、Vector分离架构,返回AI Core的核数
针对Atlas 推理系列产品(Ascend 310P处理器),非Cube、Vector分离架构,返回AI Core的核数
Atlas A2训练系列产品/Atlas 800I A2推理产品,Cube、Vector分离架构,返回AI Core上的Vector核数
约束说明
无
调用示例
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; }
父主题: PlatformAscendC类