kernels.double.binomial.cu Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nd4j-jcublas-common Show documentation

There is a newer version: 0.4-rc3.7

#include 
#include 
__device__ double doBinomial(int n, double p, double *randomNumbers,curandGenerator_t s) {
  int x = 0;
  int tid = threadIdx.x + blockIdx.x * blockDim.x;

  for(int i = tid; i < n; i++) {
    if(randomNumbers[i]< p )
      x++;
  }
  return x;
}


extern "C"
__global__ void binomial_double(int len,int n,double *ps,double *randomNumbers,double *result, curandGenerator_t s) {
          int tid = threadIdx.x + blockIdx.x * blockDim.x;
          for(int i = tid; i < len; i += blockDim.x * gridDim.x) {
              result[i] = doBinomial(n,ps[i],randomNumbers,s);
          }
}