kernels.float.min_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 "reduce.h"
__device__ float update(float old,float opOutput,float *extraParams) {
        return fminf(old,opOutput);
 }

__device__ float merge(float old,float opOutput,float *extraParams) {
        return fminf(old,opOutput);
 }

/**
 An op on the device
 @param d1 the first operator
 @param d2 the second operator
*/
__device__ float op(float d1,float d2,float *extraParams) {
      return d1;
}

__device__ float op(float d1,float *extraParams) {
      return d1;
}



__device__ float postProcess(float reduction,int n,int xOffset,float *dx,int incx,float *extraParams,float *result) {
             return reduction;
}
extern "C"
__global__ void min_strided_float(int n, int xOffset,float *dx,int incx,float *extraParams,float *result) {
             transform(n,xOffset,dx,incx,extraParams,result);
}