获取当前任务配置的核数,用于代码内部的多核逻辑控制等。
__aicore__ inline int64_t GetBlockNum()
无
当前任务配置的核数。
Kirin9020系列处理器
Kirin9030系列处理器
KirinX90系列处理器
#include "kernel_operator.h" // 在核内做简单的tiling计算时使用block_num,复杂tiling建议在host侧完成 __aicore__ inline void InitTilingParam(int32_t& totalSize, int32_t& loopSize) { loopSize = totalSize / AscendC::GetBlockNum(); };