CUDA 数学函数

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。