SetStoreAtomicConfig(ISASI)

产 品 支 持 情 况

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

功 能 说 明

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

设 置 原 子 操 作 开 启 位 与 原 子 操 作 类 型。

函 数 原 型

C++

template <AtomicDtype type, AtomicOp op>
__aicore__ inline void SetStoreAtomicConfig()

参 数 说 明

表 1 模 板 参 数 说 明

参 数 名	输 入/输 出	描 述
type	输 入	原 子 操 作 开 启 位，AtomicDtype枚 举 类 的 定 义 如 下： enum class AtomicDtype { ATOMIC_NONE = 0, // 无 原 子 操 作 ATOMIC_F32, // 开 启 原 子 操 作，进 行 原 子 操 作 的 数 据 类 型 为float ATOMIC_F16, // 开 启 原 子 操 作，进 行 原 子 操 作 的 数 据 类 型 为half ATOMIC_S16, // 开 启 原 子 操 作，进 行 原 子 操 作 的 数 据 类 型 为int16_t ATOMIC_S32, // 开 启 原 子 操 作，进 行 原 子 操 作 的 数 据 类 型 为int32_t ATOMIC_S8, // 开 启 原 子 操 作，进 行 原 子 操 作 的 数 据 类 型 为int8_t ATOMIC_BF16 // 开 启 原 子 操 作，进 行 原 子 操 作 的 数 据 类 型 为bfloat16_t };
op	输 入	原 子 操 作 类 型，仅 当 开 启 原 子 操 作 时 有 效（即"type"为 非"ATOMIC_NONE"的 场 景），当 前 仅 支 持 求 和 操 作。 enum class AtomicOp { ATOMIC_SUM = 0 // 求 和 操 作 };

数 据 类 型

支 持 的 数 据 类 型 为int8_t、int16_t、half、bfloat16_t、int32_t、float。

返 回 值 说 明

无

约 束 说 明

无

调 用 示 例

C++

// 设 置 原 子 操 作 为 求 和 操 作，支 持 的 数 据 类 型 为half
AscendC::SetStoreAtomicConfig<AscendC::AtomicDtype::ATOMIC_F16, AscendC::AtomicOp::ATOMIC_SUM>();