kernels.complex.double.sign.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

extern "C"
#include 

__global__ void sign_double(int n,int idx,double *dy,int incy,double *result) {
           for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < n; i += blockDim.x * gridDim.x) {
                          if(i >= idx && i % incy == 0) {
                              double x = dy[i];
                              result[i] =  (x > 0) - (x < 0);
                           }
            }

    }