GetTaskRatio
产品支持情况
- Ascend 950PR/Ascend 950DT:支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
- Atlas 200I/500 A2 推理产品:不支持
- Atlas 推理系列产品AI Core:支持
- Atlas 推理系列产品Vector Core:不支持
- Atlas 训练系列产品:不支持
- Kirin X90:支持
- Kirin 9030:支持
功能说明
头文件路径为:"basic_api/kernel_operator_sys_var_intf.h"。
分离模式下,获取任务启动的Cube Core(AIC)或者Vector Core(AIV)的数量与逻辑AI Core数量的比例。
- 在AIC上调用时返回AIC数量与逻辑AI Core数量的比例。
- 在AIV上调用时返回AIV数量与逻辑AI Core数量的比例。
耦合模式下,固定返回1。
函数原型
C++
__aicore__ inline int64_t GetTaskRatio()
参数说明
无
返回值说明
针对分离模式,不同Kernel类型下(通过设置Kernel类型接口设置),在AIC和AIV上调用该接口的返回值如下:
表1 返回值列表
Kernel类型 KERNEL_TYPE_AIV_ONLY KERNEL_TYPE_AIC_ONLY KERNEL_TYPE_MIX_AIC_1_2 KERNEL_TYPE_MIX_AIC_1_1 KERNEL_TYPE_MIX_AIC_1_0 KERNEL_TYPE_MIX_AIV_1_0 AIV 1 - 2 1 - 1 AIC - 1 1 1 1 - 针对耦合模式,固定返回1。
约束说明
无
调用示例
C++
int64_t ratio = AscendC::GetTaskRatio(); // 返回AIC或AIV与逻辑AI Core数量的比例
AscendC::printf("task ratio is %ld", ratio);