• Home
• org.nd4j
• nd4j-jcublas-common
• 0.0.3.5.5.3
• source code
• eps_strided.cu

×

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.double.eps_strided.cu Maven / Gradle / Ivy

#include "pairwise_transform.h"

#define MIN 1e-12


__device__ double op(double d1,double d2,double *params) {
      double diff = d1 - d2;
      double absDiff = abs(diff);
      if(absDiff < MIN)
            return 1;
       return 0;
}
__device__ double op(double d1,double *params) {
         return d1;
}

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

 }