GetTaskRation
功能说明
适用于Cube/Vector分离架构,用来获取Cube/Vector的配比。
函数原型
__aicore__ inline int64_t GetTaskRation()
参数说明
无
返回值
Cube/Vector的配比:
- 针对分离架构,该接口返回算子执行时block(逻辑概念)内的Cube/Vector核数量。融合算子的1个block内通常有1个Cube+2个Vector,该接口在Cube上会返回block内Cube核数(通常为1),在Vector上会返回Vector核数(通常为2)。对于纯Vector/纯Cube算子,1个block内只有1个Vector核/Core核,此时该接口固定返回1;
- 针对耦合架构,固定返回1。
支持的型号
Atlas推理系列产品AI Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
约束说明
无
调用示例
uint64_t ratio = GetTaskRation(); // 在分离架构融合算子场景,可以查看Cube/Vector的配比 PRINTF("task ration is %u", ratio);
父主题: 系统变量访问