cuda原子操作

来源:互联网 发布:c语言中的getchar 编辑:程序博客网 时间:2024/06/06 03:36


编译时记得 在nvcc后面加上参数"-arch=sm_12"指定计算能力。原子函数只能被计算能力1.1以上的设备支持,但是其中对于共享存储器的原子操作,和对全局存储器64bit的,只有计算能力1.2以上才支持。因此要在编译时制定,否则会报找不到atomicAdd 的错误。
C.1  数学函数C.1.1  atomicAdd()

int atomicAdd(int* address, int val);

unsigned int atomicAdd(unsigned int* address,

                           unsigned int val);

unsigned long long int atomicAdd(unsigned long long int* address,

                                        unsigned long long int val);

读取位于全局或共享存储器中地址address 处的32 位或64 位字old,计算(old + val),并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

只有全局存储器支持64 位字。

C.1.2  atomicSub()

int atomicSub(int* address, int val);

unsigned int atomicSub(unsigned int* address,

                           unsigned int val);

读取位于全局或共享存储器中地址address 处的32 位字old,计算(old - val),并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

C.1.3  atomicExch()

int atomicExch(int* address, int val);

unsigned int atomicExch(unsigned int* address,

                            unsigned int val);

unsigned long long int atomicExch(unsigned long long int* address,

                                        unsigned long long int val);

float atomicExch(float* address, float val);

读取位于全局或共享存储器中地址address 处的32 位或64 位字old,并将val 存储在存储器的同一地址中。这两项操作在一次原子事务中执行。该函数将返回old

只有全局存储器支持64 位字。

C.1.4  atomicMin()

int atomicMin(int* address, int val);

unsigned int atomicMin(unsigned int* address,

                           unsigned int val);

读取位于全局或共享存储器中地址address 处的32 位字old,计算old val 的最小值,并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

C.1.5  atomicMax()

int atomicMax(int* address, int val);

unsigned int atomicMax(unsigned int* address,

                           unsigned int val);

读取位于全局或共享存储器中地址address 处的32 位字old,计算old val 的最大值,并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

C.1.6  atomicInc()

unsigned int atomicInc(unsigned int* address,

                       unsigned int val);

读取位于全局或共享存储器中地址address 处的32 位字old,计算 ((old >= val) ? 0 : (old+1)),并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

C.1.7  atomicDec()

unsigned int atomicDec(unsigned int* address,

                           unsigned int val);

读取位于全局或共享存储器中地址address 处的32 位字old,计算 (((old == 0) | (old > val)) ? val : (old-1)),并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

C.1.8  atomicCAS()

int atomicCAS(int* address, int compare, int val);

unsigned int atomicCAS(unsigned int* address,

                           unsigned int compare,

                           unsigned int val);

unsigned long long int atomicCAS(unsigned long long int* address,

                                       unsigned long long int compare,

                                       unsigned long long int val);

读取位于全局或共享存储器中地址address 处的32 位或64 位字old,计算 (old == compare ? val : old),并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old(比较并交换)。

只有全局存储器支持64 位字。

C.2  位逻辑函数C.2.1  atomicAnd()

int atomicAnd(int* address, int val);

unsigned int atomicAnd(unsigned int* address,

                           unsigned int val);

读取位于全局或共享存储器中地址address 处的32 位字old,计算 (old & val),并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

C.2.2  atomicOr()

int atomicOr(int* address, int val);

unsigned int atomicOr(unsigned int* address,

                          unsigned int val);

读取位于全局或共享存储器中地址address 处的32 位字old,计算 (old | val),并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

C.2.3  atomicXor()

int atomicXor(int* address, int val);

unsigned int atomicXor(unsigned int* address,

                           unsigned int val);

读取位于全局或共享存储器中地址address 处的32 位字old,计算 (old ^ val),并将结果存储在存储器的同一地址中。这三项操作在一次原子事务中执行。该函数将返回old

原创粉丝点击