• Home
• org.nd4j
• nd4j-jocl-common
• 0.4-rc0
• source code
• normal.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.normal.kh Maven / Gradle / Ivy

#include 
#include 
#include


extern "C"
__kernel void normal_float(int n,float *means,float std,float *result, curandGenerator_t s) {
        int tid = get_local_id(0) + get_group_id(0) * get_local_size(0);
         for(int i = tid; i < n; i += get_local_size(0)*get_num_groups(0)) {
                     float *curr = &result[i];
                     curandGenerateNormal(s,curr,1,means[i],std);
          }
}