kernels.float.binomial_scalar.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_scalar_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,randomNumbers,s);
          }
}