CANN内存屏障同步控制函数文档

asc_mem_bar【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√功能说明头文件路径c_api/reg_compute/reg_vector.h。Reg计算宏函数内不同流水线之间的同步指令。该同步指令指定源流水线和目的流水线如下图所示目的流水线将等待源流水线上所有指令完成才进行执行。 读写场景下当读指令使用的寄存器和写指令使用的寄存器相同时可以触发寄存器保序指令将会按照代码顺序执行不需要插入同步指令而当使用的寄存器不同时如果要确保读写指令执行则需要插入同步指令。写写场景同理。函数原型__simd_callee__ inline void asc_mem_bar(MEM_TYPE mem_type)参数说明表1参数说明 | 参数名 | 输入/输出 | 描述 | | :----- | :------- | :------- | | mem_type | 输入 | 同步流水线的类型类型为MEM_TYPE取值范围见表2 mem_type取值说明。 |表2 mem_type取值说明源流水线/目的流水线表示的含义见表3 Reg计算流水线说明 | 值 | 源流水线 | 目的流水线 | | ------ | -------- | -------- | | VV_ALL | VEC_ALL | VEC_ALL | | VST_VLD | VEC_STORE | VEC_LOAD | | VLD_VST | VEC_LOAD | VEC_STORE | | VST_VST | VEC_STORE | VEC_STORE | | VS_ALL | VEC_ALL | SCALAR_ALL | | VST_LD | VEC_STORE | SCALAR_LOAD | | VLD_ST | VEC_LOAD | SCALAR_STORE | | VST_ST | VEC_STORE | SCALAR_STORE | | SV_ALL | SCALAR_ALL | VEC_ALL | | ST_VLD | SCALAR_STORE | VEC_LOAD | | LD_VST | SCALAR_LOAD | VEC_STORE | | ST_VST | SCALAR_STORE | VEC_STORE | | SS_ALL | SCALAR_ALL | SCALAR_ALL | | ST_LD | SCALAR_STORE | SCALAR_LOAD | | LD_ST | SCALAR_LOAD | SCALAR_STORE | | ST_ST | SCALAR_STORE | SCALAR_STORE |表3 Reg计算流水线说明 | 流水线 | 含义 | | ------ | -------- | | VEC_STORE | SIMD_VF函数内矢量写UB流水线。对应寄存器到UB的搬运指令例如asc_store。 | | VEC_LOAD | SIMD_VF函数内矢量读UB流水线。对应UB到寄存器的搬运指令例如asc_load。 | | SCALAR_STORE | SIMD_VF函数内标量写UB流水线。对应标量写入UB的指令例如asc_duplicate_scalar。 | | SCALAR_LOAD | SIMD_VF函数内标量读UB流水线。对应UB读取标量的指令。 | | VEC_ALL | SIMD_VF函数内所有矢量读写UB流水线。 | | SCALAR_ALL | SIMD_VF函数内所有标量读写UB流水线。 |返回值说明无流水类型PIPE_S约束说明无调用示例如下示例中for循环中后一次循环中读UB矢量搬运和前一次循环中写UB矢量搬运操作了同一块UB地址空间。因此后一次循环中读UB矢量搬运需要等待前一次循环中写UB矢量搬运执行完成后才能执行需要插入VEC_LOAD等待VEC_STORE的同步VST_VLD。// ub_addr是外部输入的UB内存空间地址 __ubuf__ half* ub_addr; vector_half dst; vector_half src0; vector_half src1; vector_bool mask; mask asc_create_mask_b16(PAT_ALL); uint16_t repeatTime 8; uint16_t repeatSize 64; for (uint16_t i 1; i repeatTime; i) { asc_mem_bar(VST_VLD); asc_loadalign(src0, ub_addr); asc_loadalign(src1, ub_addr i * repeatSize); asc_add(dst, src0, src1, mask); asc_storealign(ub_addr, dst, mask); }【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考