NotifyNextBlock

功能说明

通过写GM地址,通知下一个核当前核的操作已完成,下一个核可以进行操作。使用接口前,请确保已经调用InitDetermineComputeWorkspace接口,初始化共享内存。

函数原型

1
__aicore__ inline void NotifyNextBlock(GlobalTensor<int32_t>& gmWorkspace, LocalTensor<int32_t>& ubWorkspace)

参数说明

表1 接口参数说明

参数名称

输入/输出

含义

gmWorkspace

输入

临时空间,通过写gmWorkspace通知其他核当前核已执行完成,其他核可以继续往下执行,类型为GlobalTensor。

ubWorkspace

输入

临时空间,用于操作gmWorkspace,类型为LocalTensor。

返回值

无。

支持的型号

Atlas推理系列产品AI Core

Atlas A2训练系列产品/Atlas 800I A2推理产品

约束说明

调用示例

调用样例请参考12.1.7-调用示例