kernels.float.eps_strided.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 "pairwise_transform.h"


__device__ float op(float d1,float d2,float *params) {
      float diff = d1 - d2;
      float absDiff = fabsf(diff);
      if(absDiff < 0.1)
            return 1;
       return 0;
}
__device__ float op(float d1,float *params) {
         return d1;
}

extern "C"
__global__ void eps_strided_float(int n, int xOffset,int yOffset,float *dx, float *dy,int incx,int incy,float *params,float *result) {
        transform(n,xOffset,yOffset,dx,dy,incx,incy,params,result);

 }