CANN/asc-devkit禁用DMA原子操作
DisableDmaAtomic【免费下载链接】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 训练系列产品支持功能说明头文件路径为basic_api/kernel_operator_set_atomic_intf.h。关闭数据搬运随路原子操作功能后续执行数据搬运时GM中原始数据将被新搬运数据完全覆盖。函数原型__aicore__ inline void DisableDmaAtomic()参数说明无返回值说明无约束说明SetAtomicAdd、SetAtomicMax.md)、SetAtomicMin.md)使用完成后通过DisableDmaAtomic清空原子操作的状态以免影响后续相关指令功能。调用示例AscendC::LocalMemAllocatorAscendC::Hardware::UB ubAllocator; AscendC::LocalTensorT srcLocal ubAllocator.AllocT, SIZE(); AscendC::DisableDmaAtomic(); AscendC::DataCopy(srcLocal, srcGlobal, SIZE); AscendC::SetFlagAscendC::HardEvent::MTE2_MTE3(EVENT_ID0); AscendC::WaitFlagAscendC::HardEvent::MTE2_MTE3(EVENT_ID0); AscendC::SyncAll(); // 开启原子累加将UB数据原子累加到GM AscendC::SetAtomicAddT(); AscendC::DataCopy(dstGlobal, srcLocal, SIZE); // 关闭原子累加 AscendC::DisableDmaAtomic();完整样例请参考DataMovementWithAtomicOperations样例。【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
