原创作品,如要转载请注明出处:http://www.cnblogs.com/shrimp-can/p/6590152.html
最近要在device函数中使用curand库生成随机数,查找了下资料,除了NVIDIA官网的介绍外,其他介绍deviceAPI的我没有找到,倒是介绍hostAPI的一大堆,因此简单整理下自己所理解的deviceAPI的使用。
一、简介
curand由两部分组成:host端的库和device端的头文件。
host端的库就像其他的CPU库一样curand.h,随机数可以在设备端生成或者CPU端生成。在设备端生成的时候,对库的调用发生在host端,但是随机数的生成实际上发生在device端,随机数存储在global memory中,用户可以调用kernel直接使用这个随机数,也可以将此随机数拷贝回kernel。
device端的库头文件curand_kernel.h,这个头文件里面定义了设置随机数生成器的状态以及生成一系列随机数的设备函数,使得kernel可以调用函数生成随机数,而不需要从global memory处读写。
二、device API
包含头文件curand_kernel.h,工程需要链接到库curand。
使用curand的deviceAPI生成随机数主要需要三个步骤
a.创建一个随机算法状态的对象,如curandState state
b.对状态进行初始化,使用curand_init()函数
c.生成随机数,使用curand等
deviceAPI的作用也是这些,下面进行更详细的介绍
1.创建随机算法状态对象
在deviceAPI中,有4中为随机算法,4种真随机算法
伪随机算法 对应的状态
Mtgp32 curandStateMtgp32_t
MRG32k3a curandStateMRG32k3a_t
Philox4_32_10 curandStatePhilox4_32_10_t
XORWOW curandStateWORWOW_t
真随机算法 对应的状态
scrambled Sobol64 curandStateScrambledSobol64_t
Sobol64 curandStateSobol64_t
scrambled Sobol32 curandStateScrambledSobol32_t
Sobol32 curandStateSobol32_t
如要创建算法XORWOW状态对象:curandStateWORWOW_t state
2.对算法状态进行初始化
curand_init()函数,不同的算法,函数的参数不相同,具体算法所需要的参数可以看NVIDIA官网curand_initAPI:http://docs.nvidia.com/cuda/curand/group__DEVICE.html#group__DEVICE
对于除了MTGP32的伪随机算法,初始化参数都是:种子、序列、偏移、要初始化的随机算法状态对象
注意:算法MTGP32的初始化是通过两个hostAPI,curandMakeMTGP32Constants和cirandMakeMTGP32KernelState
如要初始化算法XORWOW状态对象:curand_init ( unsigned long long seed, unsigned long long subsequence, unsigned long long offset, curandStateXORWOW_t* state )
3.生成随机数
生成随机数除了可以使用不同的算法(由参数决定)以外,还能生成不同分布、不同浮点类型的随机数,这根据调用的生成随机数的API决定。
主要有以下几种API:
curand 生成unsigned int型整数,没有特殊分布
curand_uniform 生成服从均匀分布的float
curand_uniform4 生成4个服从均匀分布的float
curand_uniform_double 生成服从均匀分布的double
curand_uniform2_double 生成2个服从均匀分布的double
curand_poisson 生成服从泊松分布的int
curand_poisson4 生成4个服从泊松分布的int
curand_normal 生成服从正态分布的float
curand_normal2 生成2个服从正态分布的float
curand_normal4 生成4个服从正态分布的float
curand_normal_double 生成服从正态分布的double
curand_normal2_double 生成2个服从正态分布的double
curand_log_normal 生成服从对数正态分布的float
curand_log_normal2 生成2个服从对数正态分布的float
curand_log_normal4 生成4个服从对数正态分布的float
curand_log_normal_double 生成服从对数正态分布的double
curand_log_normal2_double 生成2个服从对数正态分布的double
curand_mtgp32_single 生成mtgp32算法中服从均匀分布的float
curand_mtgp32_single_specific 生成mtgp32算法中服从特殊泊松分布的float
curand_mtgp32_specific 生成mtgp32算法中的32bits的int
如要生成一个int型的随机数,采用XORWOW算法:curand(curandStateXORWOW_t* state)
以上是个人理解,如有错误的地方,忘批评指正
urand_mtgp32_single 生成mtgp32算法中服从均匀分布的floa
curand_log_normal2 生成2个服从对数正态分布的float_log_normal 生成服从对数正态分布的floatcurand_log_normal 生成服从对数正态分布的floatcu
curand_log_normal 生成服从对数正态分布的floa
curand_log_normal 生成服从对数正态分布的floacurand_log_normal 生成服从对数正态分布的floacurand_log_normal 生成服从对数正态分布的f