跳到主要内容

GetBlockNum

功能说明​

获取当前任务配置的核数,用于代码内部的多核逻辑控制等。

函数原型​

__aicore__ inline int64_t GetBlockNum()

参数说明​

无

返回值​

当前任务配置的核数。

支持的型号​

Kirin9020系列处理器

KirinX90系列处理器

约束说明​

无

调用示例​

#include "kernel_operator.h"
// 在核内做简单的tiling计算时使用block_num,复杂tiling建议在host侧完成
__aicore__ inline void InitTilingParam(int32_t& totalSize, int32_t& loopSize)
{
loopSize = totalSize / AscendC::GetBlockNum();
};