CANN Ascend C原子操作类型设置

SetAtomicType【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品支持Atlas A2 训练系列产品/Atlas A2 推理系列产品支持Atlas 200I/500 A2 推理产品支持Atlas 推理系列产品AI Core支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明头文件路径为basic_api/kernel_operator_set_atomic_intf.h。通过设置模板参数来设定原子操作不同的数据类型。函数原型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.md)、SetAtomicMin.md)内部已集成与SetAtomicType相同的功能。建议调用上述接口时通过设置模板参数显式指定原子操作的数据类型无需额外调用SetAtomicType接口。使用完成后建议使用DisableDmaAtomic清空原子操作的状态以免影响后续相关指令功能。调用示例根据约束说明可知使用SetAtomicAdd、SetAtomicMax.md)、SetAtomicMin.md)时已通过模板参数指定原子操作的数据类型无需额外调用SetAtomicType接口因此本接口不提供调用示例。【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考