GetBlockNum
功能说明
获取当前任务配置的核数,用于代码内部的多核逻辑控制等。
函数原型
1
|
__aicore__ inline int64_t GetBlockNum() |
参数说明
无
返回值
当前任务配置的核数。
支持的型号
Atlas 训练系列产品
Atlas推理系列产品AI Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
Atlas 200/500 A2推理产品
约束说明
无。
调用示例
1 2 3 4 5 6 |
#include "kernel_operator.h" // 在核内做简单的tiling计算时使用block_num,复杂tiling建议在host侧完成 __aicore__ inline void InitTilingParam(int32_t& totalSize, int32_t& loopSize) { loopSize = totalSize / AscendC::GetBlockNum(); }; |
父主题: 系统变量访问