asc_sync_block_arrive

产品支持情况

产品 是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Ascend 950PR/Ascend 950DT

功能说明

asc_sync_block_wait配合使用(通过flag_id关联),用于组内block和subblock间的信号同步。通过向核间同步寄存器发送同步信号,标识当前pipeline已到达flag_id对应的同步点。每个flag_id对应一个初始值为0的计数器,执行asc_sync_block_arrive后flag_id对应的计数器增加1;执行asc_sync_block_wait时如果对应的计数器数值为0则阻塞不执行;如果对应的计数器大于0,则计数器减一,同时后续指令开始执行。

函数原型

__aicore__ inline void asc_sync_block_arrive(pipe_t pipe, int64_t flag_id)

参数说明

参数名 输入/输出 描述
pipe 输入 设置这条指令所在的流水类型。
flag_id 输入 事件标号,用于标识同一组同步信号。取值范围为[0,15]。

返回值说明

流水类型

PIPE_S

约束说明

  • 使用该同步接口时,需要按照如下规则设置Kernel类型:
    • 在纯Vector/Cube场景下,需设置Kernel类型为KERNEL_TYPE_MIX_AIV_1_0或KERNEL_TYPE_MIX_AIC_1_0。
    • 对于Vector和Cube混合场景,需根据实际情况灵活配置Kernel类型。
  • 不建议开发者同时使用该接口和Matmul高阶API,否则会有flag_id冲突的风险。
  • 同一flag_id的计数器最多设置15次。

调用示例

// Host侧调用接口aclrtGetHardwareSyncAddr获取核间同步基地址ffts_addr
uint64_t config = *(__gm__ uint64_t*)ffts_addr;
asc_set_ffts_base_addr(config);
int64_t flag_id = 1;
asc_sync_block_arrive(PIPE_S, flag_id);
asc_sync_block_wait(PIPE_S, flag_id);  
// 在Ascend 950PR/Ascend 950DT产品上,无需进行核间同步基地址ffts_addr的设置
int64_t flag_id = 1;
asc_sync_block_arrive(PIPE_S, flag_id);
asc_sync_block_wait(PIPE_S, flag_id);