• Home
• org.nd4j
• nd4j-jocl-common
• 0.4-rc3.6
• source code
• binomial.kh

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

kernels.float.binomial.kh Maven / Gradle / Ivy

#include 
#include 
__global  float doBinomial(int n, float p, float *randomNumbers,curandGenerator_t s) {
  int x = 0;
  int tid = get_local_id(0) + get_group_id(0) * get_local_size(0);

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


extern "C"
__kernel void binomial_float(int len,int n,float *ps,float *randomNumbers,float *result, curandGenerator_t s) {
          int tid = get_local_id(0) + get_group_id(0) * get_local_size(0);
          for(int i = tid; i < len; i += get_local_size(0) * get_num_groups(0)) {
              result[i] = doBinomial(n,ps[i],randomNumbers,s);
          }
}