CANN原子递减操作API文档

📅 2026/7/16 18:31:46
CANN原子递减操作API文档
asc_atomic_dec【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品不支持Atlas A2 训练系列产品/Atlas A2 推理系列产品不支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core不支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明头文件路径c_api/scalar_compute/scalar_compute.h。对Global Memory中address指向的计数器执行原子递减操作如果address上的数值等于0或大于指定数值val则对address赋值为val否则将address上数值减1。函数原型__aicore__ inline uint32_t asc_atomic_dec(__gm__ uint32_t *address, uint32_t val) __aicore__ inline uint64_t asc_atomic_dec(__gm__ uint64_t *address, uint64_t val)参数说明参数名输入/输出描述address输出Global Memory的地址。val输入源操作数。返回值说明address地址中计算前的原始数据。流水类型PIPE_S约束说明在开启编译器自动同步功能的前提下编译器能够自动在PIPE_MTE2/PIPE_MTE3与PIPE_S之间插入同步。但是asc_atomic_dec为标量计算在读写GM时如果与搬运单元MTE2/MTE3存在数据依赖编译器却无法自动插入同步开发者需要根据实际情况手动插入同步。Scalar原子操作会绕过DCache需要调用asc_dcci接口确保GM与DCache的一致性。调用示例uint32_t counter 5; // 初始值 asc_dcci_entire_all(); // 手动同步 uint32_t old asc_atomic_dec(counter, 9); // counter 4, old 5【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考