Skip to content
版 本

GetStoreAtomicConfig(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"

获 取 原 子 操 作 开 启 位 与 原 子 操 作 类 型 的 值,详 细 说 明 见表1

函 数 原 型

C++
__aicore__ inline void GetStoreAtomicConfig(uint16_t& atomicType, uint16_t& atomicOp)

参 数 说 明

表 1 参 数 说 明

参 数 名输 入/输 出描 述
atomicType输 出原 子 操 作 开 启 位。
0:无 原 子 操 作
1:开 启 原 子 操 作,进 行 原 子 操 作 的 数 据 类 型 为float
2:开 启 原 子 操 作,进 行 原 子 操 作 的 数 据 类 型 为half
3:开 启 原 子 操 作,进 行 原 子 操 作 的 数 据 类 型 为int16_t
4:开 启 原 子 操 作,进 行 原 子 操 作 的 数 据 类 型 为int32_t
5:开 启 原 子 操 作,进 行 原 子 操 作 的 数 据 类 型 为int8_t
6:开 启 原 子 操 作,进 行 原 子 操 作 的 数 据 类 型 为bfloat16_t
atomicOp输 出原 子 操 作 类 型。
0:求 和 操 作

返 回 值 说 明

约 束 说 明

此 接 口 需 要 与SetStoreAtomicConfig(ISASI)配 合 使 用,用 以 获 取 原 子 操 作 开 启 位 与 原 子 操 作 类 型 的 值。

调 用 示 例

C++
AscendC::SetStoreAtomicConfig<AscendC::AtomicDtype::ATOMIC_F16, AscendC::AtomicOp::ATOMIC_SUM>();
uint16_t type = 0;       // 原 子 操 作 开 启 位
uint16_t op = 0;         // 原 子 操 作 类 型
AscendC::GetStoreAtomicConfig(type, op);

免 责 声 明:本 站 内 容 由 asc-devkit 仓 master 分 支 自 动 编 译 生 成,属 于 持 续 开 发 版 本,可 能 存 在 缺 陷,仅 供 预 览 与 参 考。如 需 稳 定 及 商 用 资 料,请 查 阅 官 方 昇 腾 社 区