Skip to content

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_ONLYKERNEL_TYPE_AIC_ONLYKERNEL_TYPE_MIX_AIC_1_2KERNEL_TYPE_MIX_AIC_1_1KERNEL_TYPE_MIX_AIC_1_0KERNEL_TYPE_MIX_AIV_1_0
    AIV1-21-1
    AIC-1111-
  • 针对耦合模式,固定返回1。

约束说明

调用示例

C++
int64_t ratio = AscendC::GetTaskRatio(); // 返回AIC或AIV与逻辑AI Core数量的比例
AscendC::printf("task ratio is %ld", ratio);

免责声明:本站内容由 asc-devkit 仓 master 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区