昇腾社区首页
中文
注册
开发者
下载

Atomic原子操作

功能说明

set_atomic_add()函数功能:调用该接口后,可对后续的从L0C/L1/UB移动到GM的数据传输开启原子累加;set_atomic_s8/s16/bf16/f16/s32/f32()函数指定写入的数据类型。

set_atomic_max()函数功能:调用该接口后,将待拷贝的内容和GM已有内容进行比较,将最大值写入GM;set_atomic_s8/s16/bf16/f16/s32/f32()函数指定写入的数据类型。

set_atomic_min()函数功能:调用该接口后,将待拷贝的内容和GM已有内容进行比较,将最小值写入GM;set_atomic_s8/s16/bf16/f16/s32/f32()函数指定写入的数据类型。

set_atomic_none()函数功能:清空原子操作的状态。

set_atomic_s8/s16/bf16/f16/s32/f32()函数: 用于指定写入的数据类型。

接口原型

void set_atomic_add(); // 与搬运的目的地址数据进行AtomicAdd操作
void set_atomic_max(); // 与搬运的目的地址数据进行AtomicMax操作
void set_atomic_min(); // 与搬运的目的地址数据进行AtomicMin操作

// 设置Atomic写入的不同数据类型
void set_atomic_s8();
void set_atomic_s16();
void set_atomic_bf16();
void set_atomic_f16();
void set_atomic_s32();
void set_atomic_f32();

// 重置,不再进行Atomic操作
void set_atomic_none(); 

流水类型

PIPE_S