GetStoreAtomicConfig
功能说明
获取原子操作使能位与原子操作类型的值,详细说明见表1。
函数原型
__aicore__ inline void GetStoreAtomicConfig(uint16_t &atomicType, uint16_t &atomicOp)
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
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);
父主题: 原子操作