cann/asc-devkit GetTaskRatio API文档

GetTaskRatio

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

产品支持情况

• 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。

函数原型

__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。

约束说明

无

调用示例

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

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit