Cuda生成随机数的一点浅薄的理解

老实人小李

于 2021-11-02 20:27:58 发布

阅读量372

点赞数

分类专栏： CUDA 文章标签： cuda

本文链接：https://blog.csdn.net/weixin_43660703/article/details/121108247

版权

CUDA 专栏收录该内容

5 篇文章 1 订阅

订阅专栏

一个函数内，相同种子的多次生成


#include "cuda_runtime.h"
#include "device_launch_parameters.h"
#include "curand_kernel.h"

#include <stdio.h>

#define N 100
#define BLOCKS 2
#define THREAD_NUM 2

__global__ void setup_kernel(curandState *state, unsigned long seed)
{
	int tid = blockIdx.x *blockDim.x + threadIdx.x; //获取线程号0~blocks*THREAD_NUM-1  grid划分成1维，block划分为1维
	curand_init(seed, tid, 0, &state[tid]);// initialize the state
}

__global__ void use(curandState *globalState)
{
	//unsigned int j;
	int j;
	int tid = blockIdx.x *blockDim.x + threadIdx.x; //获取线程号0~blocks*THREAD_NUM-1  grid划分成1维，block划分为1维
	curandState localState = globalState[tid];
	for (int i = 0; i < 5; i++)
	{
		j = (curand(&localState)) % N;
		//printf("%u\n", j);
		//printf("%d\n", j);
		printf("%d\n", j);
	}
}

int main()
{
	curandState* devStates;  //创建一个随机算法状态的对象
	cudaMalloc(&devStates, BLOCKS * THREAD_NUM * sizeof(curandState));
	srand(time(0));

	setup_kernel << <BLOCKS, THREAD_NUM >> > (devStates, rand()); // blocks number is POP. thread number is P
	use << < BLOCKS, THREAD_NUM >> > (devStates);
    
    return 0;
}