通过写GM地址,通知下一个核当前核的操作已完成,下一个核可以进行操作。使用接口前,请确保已经调用InitDetermineComputeWorkspace接口,初始化共享内存。
1
|
__aicore__ inline void NotifyNextBlock(GlobalTensor<int32_t>& gmWorkspace, LocalTensor<int32_t>& ubWorkspace) |
参数名称 |
输入/输出 |
含义 |
---|---|---|
gmWorkspace |
输入 |
临时空间,通过写gmWorkspace通知其他核当前核已执行完成,其他核可以继续往下执行,类型为GlobalTensor。 |
ubWorkspace |
输入 |
临时空间,用于操作gmWorkspace,类型为LocalTensor。 |
无。
Atlas推理系列产品AI Core
Atlas A2训练系列产品/Atlas 800I A2推理产品
调用样例请参考12.1.7-调用示例。