cuda 中,一个线程的原子操作可以在不受其他线程的任何操作的影响下完成对某个(全局内存或共享内存)
数据的一套“读-改-写”操作。
有两种方法能够在GPU中得到最终结果:
- 用另一个核函数将较短的数组进一步归约;
- 在核函数末尾利用原子函数进行归约。
在代码实现中:
- 原子函数
atomicAdd(·)
执行数组的一次完整的读-写操作; - 传给
cudaMemcpy(·)
的主机内存可以是栈内存,也可以是堆内存; - 主机函数可以和设备函数同名,但要遵循重载原则(参数列表不一致)。
原子函数对其第一个参数指向的数据进行一次“读-写-改”的原子操作,是不可分割的操作。
第一个参数可以指向全局内存,也可以指向共享内存。
对所有参与的线程来说,原子操作是一个线程一个线程轮流进行的,没有明确的次序。
原子函数没有同步功能。
原子函数的返回值为所指地址的旧值。
- 加法:
T atomicAdd(T *address, T val)
; - 减法:
T atomicSub(T *address, T val)
; - 交换:
T atomicExch(T *address, T val)
; - 最小值:
T atomicMin(T *address, T val)
; - 最大值:
T atomicMax(T *address, T val)
; - 自增:
T atomicInc(T *address, T val)
; - 自减:
T atomicDec(T *address, T val)
; - 比较交换:
T atomicCAS(T *address, T compare, T val)
;
两个粒子互为邻居的判断:他们的距离不大于一个给定的截断距离 rc。
基本算法: 对每一个给定的粒子,通过比较它与所有其他粒子的距离来判断相应粒子对是否互为邻居。