OpenCL中连续写入是否具备原子性？含代码场景分析

咱们一步一步来拆解你的问题：

第一个场景：独立的标量写入操作 #

先看你的第一段代码：

void set_vals(__global int* mem) { mem[0] = 1; mem[1] = 2; }
void get_vals(__global int* mem) { int x = mem[0]; int y = mem[1]; printf("%d %d\n", x, y); }

这里set_vals里是两个独立的标量存储操作：先写mem[0]，再写mem[1]。根据OpenCL的内存模型：

• 单个对齐的标量存储（比如int）是原子的——也就是说，其他工作项要么看到mem[0]的旧值（0），要么看到新值（1），不会看到中间的部分更新（因为int的存储是一次性完成的）。
• 但两个独立的存储操作之间没有原子性保证。同一个工作项内的指令是顺序执行的（mem[0]的写一定会在mem[1]的写之前完成），但另一个工作项的get_vals可以在这两个写入之间插入读取操作。

所以可能的输出包括：

• 0 0：get_vals在两个写入都未执行时读取
• 1 0：get_vals在mem[0]写完、mem[1]还没写完时读取
• 1 2：get_vals在两个写入都完成后读取

至于0 2？不可能出现。因为set_vals所在的工作项会先完成mem[0]的写入，再执行mem[1]的写入，其他工作项不可能看到mem[1]是2而mem[0]还是0的状态。

第二个场景：向量类型的写入操作 #

再看第二段代码：

void set_vals(__global int2* mem) { mem[0] = (int2)(1, 2); }
void get_vals(__global int2* mem) { int2 a = mem[0]; printf("%d %d\n", a.x, a.y); }

这里的关键是：单个对齐的向量存储操作是原子的。只要mem指向的内存满足int2的对齐要求（通常是8字节对齐，因为int2是两个int，每个4字节），这个向量写入就是不可分割的——其他工作项要么读取到完整的旧值(0,0)，要么读取到完整的新值(1,2)，绝对不会出现1 0或0 2这种部分更新的情况。

需要注意的例外情况：如果内存没有正确对齐，设备可能会把向量存储拆分成多个标量存储操作，这时候就会回到第一个场景的情况，可能出现部分更新。但只要你是正常声明和使用__global int2*，OpenCL会保证内存对齐，所以不用太担心这个问题。

总结一下：

• 多个独立的标量写入之间没有原子性，可能被其他工作项的读取打断，出现中间状态
• 单个对齐的向量写入是原子的，不会出现部分更新

内容的提问来源于stack exchange，提问作者MaiaVictor