下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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);
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词