• Home
• org.nd4j
• nd4j-jcublas-common
• 0.0.3.5.5.4
• source code
• sum_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.float.sum_strided.cu Maven / Gradle / Ivy

#include "reduce.h"


__device__ float update(float old,float opOutput,float *extraParams) {
	return opOutput + old;
}


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

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


__device__ float postProcess(float reduction,int n,int xOffset,float *dx,int incx,float *params,float *result) {
	return reduction;
}


extern "C"
__global__ void sum_strided_float(int n, int xOffset,float *dx,int incx,float *params,float *result) {
	transform(n,xOffset,dx,incx,params,result);
}