下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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();
};
搜索结果
找到“0”个结果

当前产品无相关内容

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