Tạo số ngẫu nhiên CUDA

Question

Tôi muốn tạo các số ngẫu nhiên trong hàm __device__ và giữ nó trong Bảng int của mình [500], tôi đã tìm thấy một số ví dụ, nhưng chúng đã sử dụng một số loại có tên là curandState. Tôi chỉ cần một hàm như rand() trong C++.

Answer 1

đây là mã của tôi, tôi có N3 [40000] mảng trong bộ nhớ điện thoại của mình, tôi tạo ra một số số ngẫu nhiên trong hạt nhân của tôi làm việc cho một chuỗi (tôi có nghĩa là "hạt nhân < < < 1,1 >>> .. . "), sau đó tôi sao chép nó vào N2 [40000] của tôi từ CPU, và in nó, do đó, đây là mã số

#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; 
} 

Answer 2

Bạn có thể sử dụng thư viện curand để tạo số ngẫu nhiên trong bộ nhớ thiết bị và sau đó chạy hạt nhân mà không cần phải sao chép các giá trị đó vào máy chủ.