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);