通过读GM地址中的值,确认是否需要继续等待,当GM的值满足当前核的等待条件时,该核即可往下执行,进行下一步操作。使用接口前,请确保已经调用InitDetermineComputeWorkspace接口,初始化共享内存。
1
|
__aicore__ inline void WaitPreBlock(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-调用示例。