asc_sync_subblock_wait
产品支持情况
- Ascend 950PR/Ascend 950DT:支持
- Atlas A3 训练系列产品/Atlas A3 推理系列产品:支持
- Atlas A2 训练系列产品/Atlas A2 推理系列产品:支持
- Atlas 200I/500 A2 推理产品:不支持
- Atlas 推理系列产品AI Core:不支持
- Atlas 推理系列产品Vector Core:不支持
- Atlas 训练系列产品:不支持
功能说明
在分离模式下,AIC(Cube Core)与AIV(Vector Core)按group划分,一个group由1个block(主核)和N个subblock(从核)组成,比例1:N。组内subblock间同步指同一group内不同subblock(AIV)之间的同步。
核间同步指令,每个flag_id均分配有计数器,执行asc_sync_subblock_arrive后计数器将递增;执行asc_sync_subblock_wait时如果对应的计数器数值为0则阻塞;如果对应的计数器大于0,则计数器减一,同时后续指令开始执行。
函数原型
__aicore__ inline void asc_sync_subblock_wait(pipe_t pipe, uint8_t flag_id)
__aicore__ inline void asc_sync_subblock_wait(pipe_t pipe, int64_t flag_id)
参数说明
表1 参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| pipe | 输入 | 需要执行同步指令的流水类型。 |
| flag_id | 输入 | 事件标号,取值范围[0,15]。 |
返回值说明
无
流水类型
PIPE_S
约束说明
-
flag_id的取值范围为0至15,每个flag_id的计数器范围为0至15。
-
必须保证每个flag_id的计数器不溢出,否则将引发异常。
-
必须保证相同的ID在同一时间仅被一条流水线等待。
-
必须保证配套使用asc_sync_subblock_arrive和asc_sync_subblock_wait,避免计算核一直处于阻塞阶段。
-
pipe支持的流水类型为PIPE_V、PIPE_M、PIPE_MTE1、PIPE_MTE2、PIPE_MTE3、PIPE_FIX,不支持PIPE_ALL和PIPE_S。
-
flag_id的值超出取值范围时,会截取最低4位为准。
调用示例
int64_t flag_id = 1;
asc_sync_subblock_arrive(PIPE_V, flag_id);
asc_sync_subblock_wait(PIPE_V, flag_id);