kernels.float.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__ float doBinomial(int n, float p, float *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_float(int len,int n,float *ps,float *randomNumbers,float *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);
          }
}