GetStoreAtomicConfig

功能说明

获取原子操作使能位与原子操作类型的值，详细说明见表1。

函数原型

__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：求和操作

返回值

uint16_t类型的原子操作使能位与原子操作类型的值。

支持的型号

Atlas A2训练系列产品/Atlas 800I A2推理产品

约束说明

此接口需要与SetStoreAtomicConfig配合使用，用以获取原子操作使能位与原子操作类型的值。

调用示例

SetStoreAtomicConfig<AtomicDtype::ATOMIC_F16, AtomicOp::ATOMIC_SUM>();
uint16_t type = 0;       // 原子操作使能位
uint16_t op = 0;        //原子操作类型
GetStoreAtomicConfig(type, op);

父主题： 原子操作