GetTaskRatio

产品支持情况

产品

是否支持

Ascend 950PR/Ascend 950DT

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Kirin X90

Kirin 9030

功能说明

分离模式下,获取一个AI Core上Cube Core(AIC)或者Vector Core(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。

约束说明

调用示例

// 分离模式下,获取一个AI Core上Cube Core(AIC)或者Vector Core(AIV)的数量与AI Core数量的比例。耦合模式下,固定返回1。
uint64_t ratio = AscendC::GetTaskRatio();
AscendC::PRINTF("task ratio is %u", ratio);