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

#include "scalar.h"
//scalar and current element
__device__ float op(float d1,float d2,float *params) {
       if(d2 == d1) {
       return 1;
       }
       return 0;

}

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