関数で乱数を生成し、__device__
それをint Board [500]に保持したいのですが、いくつかの例を見つけましたが、curandStateという名前の型を使用していました。rand()
C++のaのような関数だけが必要です。
13547 次
2 に答える
5
これが私のコードです。デバイスメモリにN3[40000]配列があり、1つのスレッドで動作するカーネルでいくつかの乱数を生成します(つまり、この「カーネル<<< 1,1 >>> ...」)、次に、CPUからN2 [40000]にコピーして印刷します。これが、コードです。
#include <iostream>
#include <Cuda.h>
#include<curand.h>
#include<curand_kernel.h>
int n = 200;
using namespace std;
__device__ float generate( curandState* globalState, int ind )
{
//int ind = threadIdx.x;
curandState localState = globalState[ind];
float RANDOM = curand_uniform( &localState );
globalState[ind] = localState;
return RANDOM;
}
__global__ void setup_kernel ( curandState * state, unsigned long seed )
{
int id = threadIdx.x;
curand_init ( seed, id, 0, &state[id] );
}
__global__ void kernel(float* N, curandState* globalState, int n)
{
// generate random numbers
for(int i=0;i<40000;i++)
{
int k = generate(globalState, i) * 100000;
while(k > n*n-1)
{
k-=(n*n-1);
}
N[i] = k;
}
}
int main()
{
int N=40000;
curandState* devStates;
cudaMalloc ( &devStates, N*sizeof( curandState ) );
// setup seeds
setup_kernel <<< 1, N >>> ( devStates,unsigned(time(NULL)) );
float N2[40000];
float* N3;
cudaMalloc((void**) &N3, sizeof(float)*N);
kernel<<<1,1>>> (N3, devStates, n);
cudaMemcpy(N2, N3, sizeof(float)*N, cudaMemcpyDeviceToHost);
for(int i=0;i<N;i++)
{
cout<<N2[i]<<endl;
}
return 0;
}
于 2012-09-23T10:44:44.003 に答える
3
curandライブラリを使用して、デバイスメモリに乱数を生成し、それらの値をホストにコピーしなくてもカーネルを実行できます。
于 2012-08-06T17:08:21.760 に答える