昇腾社区首页
中文
注册

SetBatchNum

功能说明

该接口为预留接口,当前版本暂不支持。

调用IterateBatch接口之前,需要在host侧Tiling实现中通过本接口设置多Batch计算的最大Batch数。

函数原型

__aicore__ inline void SetBatchNum(int32_t batchA, int32_t batchB);

参数说明

表1 参数说明

参数名

输入/输出

描述

batchA

输入

设置的一次计算的batch数。

batchB

输入

设置的一次计算的batch数。

返回值

-1表示设置失败; 0表示设置成功。

注意事项