【发布时间】:2019-08-12 03:47:01
【问题描述】:
问题How to generate random number inside pyCUDA kernel?中报道了一个CUDA或pyCUDA中随机数生成的典型例子,即
#include <curand_kernel.h>
const int nstates = %(NGENERATORS)s;
__device__ curandState_t* states[nstates];
__global__ void initkernel(int seed)
{
int tidx = threadIdx.x + blockIdx.x * blockDim.x;
if (tidx < nstates) {
curandState_t* s = new curandState_t;
if (s != 0) {
curand_init(seed, tidx, 0, s);
}
states[tidx] = s;
}
}
__global__ void randfillkernel(float *values, int N)
{
int tidx = threadIdx.x + blockIdx.x * blockDim.x;
if (tidx < nstates) {
curandState_t s = *states[tidx];
for(int i=tidx; i < N; i += blockDim.x * gridDim.x) {
values[i] = curand_uniform(&s);
}
*states[tidx] = s;
}
}
使用这个经典示例,激活的随机数生成器是什么(XORWOW、MTGP32、其他)?
如何从内核中更改随机数生成器?
【问题讨论】:
-
在 cuRand 设备 API 文档中有完整记录
-
@talonmies 据我了解,要强制
cuRAND使用Philox_4x32_10生成器,将上面示例中的curandState_t更改为curandStatePhilox4_32_10_t就足够了。