文档
注册

GetTaskRation

功能说明

适用于Cube/Vector分离架构,用来获取Cube/Vector的配比。

函数原型

__aicore__ inline int64_t GetTaskRation()

参数说明

返回值

Cube/Vector的配比:

  • 针对分离架构,该接口返回算子执行时block(逻辑概念)内的Cube/Vector核数量。融合算子的1个block内通常有1个Cube+2个Vector,该接口在Cube上会返回block内Cube核数(通常为1),在Vector上会返回Vector核数(通常为2)。对于纯Vector/纯Cube算子,1个block内只有1个Vector核/Core核,此时该接口固定返回1;
  • 针对耦合架构,固定返回1。

支持的型号

Atlas推理系列产品(Ascend 310P处理器)AI Core

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

约束说明

调用示例

    uint64_t ratio = GetTaskRation();
    // 在分离架构融合算子场景,可以查看Cube/Vector的配比
    PRINTF("task ration is %u", ratio);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词