SetAtomicType

产 品 支 持 情 况

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

功 能 说 明

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

通 过 设 置 模 板 参 数 来 设 定 原 子 操 作 不 同 的 数 据 类 型。

函 数 原 型

C++

template <typename T>
__aicore__ inline void SetAtomicType()

参 数 说 明

表 1 模 板 参 数 说 明

参 数 名	描 述
T	设 定 原 子 操 作 不 同 的 数 据 类 型。

数 据 类 型

• Ascend 950PR/Ascend 950DT，支 持 的 数 据 类 型 为int8_t、int16_t、half、bfloat16_t、int32_t、float。

• Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品，支 持 的 数 据 类 型 为int8_t、int16_t、half、bfloat16_t、int32_t、float。

• Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品，支 持 的 数 据 类 型 为int8_t、int16_t、half、bfloat16_t、int32_t、float。

• Atlas 200I/500 A2 推 理 产 品，支 持 的 数 据 类 型 为int16_t、half、int32_t、float。

• Atlas 推 理 系 列 产 品AI Core，支 持 的 数 据 类 型 为int16_t、half、float。

返 回 值 说 明

无

约 束 说 明

• SetAtomicAdd、SetAtomicMax、SetAtomicMin内 部 已 集 成 与SetAtomicType相 同 的 功 能。建 议 调 用 上 述 接 口 时，通 过 设 置 模 板 参 数 显 式 指 定 原 子 操 作 的 数 据 类 型，无 需 额 外 调 用SetAtomicType接 口。
• 使 用 完 成 后，建 议 使 用DisableDmaAtomic清 空 原 子 操 作 的 状 态，以 免 影 响 后 续 相 关 指 令 功 能。

调 用 示 例

根 据 约 束 说 明 可 知，使 用SetAtomicAdd、SetAtomicMax、SetAtomicMin时，已 通 过 模 板 参 数 指 定 原 子 操 作 的 数 据 类 型，无 需 额 外 调 用SetAtomicType接 口，因 此 本 接 口 不 提 供 调 用 示 例。