DisableDmaAtomic

产 品 支 持 情 况

产 品	是 否 支 持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品	√
Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品	√
Atlas 200I/500 A2 推 理 产 品	√
Atlas 推 理 系 列 产 品AI Core	√
Atlas 推 理 系 列 产 品Vector Core	x
Atlas 训 练 系 列 产 品	√

功 能 说 明

头 文 件 路 径 为："basic_api/kernel_operator_set_atomic_intf.h"。

关 闭 数 据 搬 运 随 路 原 子 操 作 功 能，后 续 执 行 数 据 搬 运 时，GM中 原 始 数 据 将 被 新 搬 运 数 据 完 全 覆 盖。

函 数 原 型

C++

__aicore__ inline void DisableDmaAtomic()

参 数 说 明

无

返 回 值 说 明

无

约 束 说 明

SetAtomicAdd、SetAtomicMax、SetAtomicMin使 用 完 成 后，通 过DisableDmaAtomic清 空 原 子 操 作 的 状 态，以 免 影 响 后 续 相 关 指 令 功 能。

调 用 示 例

C++

AscendC::LocalMemAllocator<AscendC::Hardware::UB> ubAllocator;
AscendC::LocalTensor<T> srcLocal = ubAllocator.Alloc<T, SIZE>();

AscendC::DisableDmaAtomic();
AscendC::DataCopy(srcLocal, srcGlobal, SIZE);
AscendC::SetFlag<AscendC::HardEvent::MTE2_MTE3>(EVENT_ID0);
AscendC::WaitFlag<AscendC::HardEvent::MTE2_MTE3>(EVENT_ID0);
AscendC::SyncAll();

// 开 启 原 子 累 加，将UB数 据 原 子 累 加 到GM
AscendC::SetAtomicAdd<T>();
AscendC::DataCopy(dstGlobal, srcLocal, SIZE);

// 关 闭 原 子 累 加
AscendC::DisableDmaAtomic();

完 整 样 例 请 参 考DataMovementWithAtomicOperations样 例。