---
title: SetAtomicAdd
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/ascendcopapi/atlasascendc_api_07_0210.html
sourcePath: /source/zh/canncommercial/900/API/ascendcopapi/atlasascendc_api_07_0210.html
indexId: e42132ccde3691d27e1a293c26bd034219aa836c6fd3b3af9e1f0a182ef91bfc76
---
# SetAtomicAdd

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | √ |
| Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 AI Core | √ |
| Atlas 推理系列产品 Vector Core | x |
| Atlas 训练系列产品 | √ |


#### 功能说明

调用该接口后，可对后续的从VECOUT/L0C/L1到GM的数据传输开启原子累加，通过模板参数设定不同的累加数据类型。


#### 函数原型

```
template <typename T>
__aicore__ inline void SetAtomicAdd()
```


#### 参数说明


**表1 模板参数说明**

| 参数名 | 描述 |
| --- | --- |
| T | 设定不同的累加数据类型。 Atlas 训练系列产品 ，支持的数据类型为：float；支持的数据通路为VECOUT\->GM。 Atlas 推理系列产品 AI Core，支持的数据类型为int16\_t/half/float；支持的数据通路为VECOUT\->GM。 Atlas A2 训练系列产品 / Atlas A2 推理系列产品 ，支持的数据类型为int8\_t/int16\_t/half/bfloat16\_t/int32\_t/float；支持的数据通路为VECOUT/L0C/L1\->GM。 Atlas A3 训练系列产品 / Atlas A3 推理系列产品 ，支持的数据类型为int8\_t/int16\_t/half/bfloat16\_t/int32\_t/float；支持的数据通路为VECOUT/L0C/L1\->GM。 Atlas 350 加速卡，支持的数据类型为int8\_t/int16\_t/half/bfloat16\_t/int32\_t/float；支持的数据通路为VECOUT/L0C Buffer\->GM。 Atlas 200I/500 A2 推理产品 ，支持的数据类型为int16\_t/half/int32\_t/float；支持的数据通路为VECOUT/L0C/L1\->GM |


#### 返回值说明

无


#### 约束说明

- 累加操作完成后，建议通过  DisableDmaAtomic
关闭原子累加，以免影响后续相关指令功能。
- 该指令执行前不会对GM的数据做清零操作，开发者根据实际的算子逻辑判断是否需要清零，如果需要自行进行清零操作。


#### 调用示例

本示例中，使用DataCopy从VECOUT搬出数据到外部dstGlobal时进行原子累加。为保证原子累加的正确性，在核函数调用前，需要对dstGm清零。

调用示例如下：

```
// 初始化LocalTensor
AscendC::LocalTensor<float> src0Local = inQueueSrc0.AllocTensor<float>();
// 清空原子操作的状态
AscendC::DisableDmaAtomic();
AscendC::DataCopy(src0Local, src0Global, 256);
// 手动插入MTE3等待src0Global搬入src0Local同步
AscendC::SetFlag<AscendC::HardEvent::MTE2_MTE3>(0);
AscendC::WaitFlag<AscendC::HardEvent::MTE2_MTE3>(0);
// 开启原子累加
AscendC::SetAtomicAdd<float>();
AscendC::DataCopy(dstGlobal, src0Local, 256);
// 清空原子操作的状态
AscendC::DisableDmaAtomic();
inQueueSrc0.FreeTensor(src0Local);
```

结果示例如下：

```
每个核的输入数据Src0: [1,1,1,1,1,...,1] // 256个1
最终输出数据dstGm: [3,3,3,3,3,...,3] // 256个3
```