• Home
• org.nd4j
• nd4j-jcublas-common
• 0.0.3.5.5.3
• source code
• sub_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.complex.float.sub_strided.cu Maven / Gradle / Ivy

extern "C"
#include 

__global__ void sub_strided_float(int n, int xOffset,int yOffset,float *dx, float *dy,int incx,int incy,float *result) {
          for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < n; i += blockDim.x * gridDim.x) {
                            if(i >= xOffset && i >= yOffset &&  i % incx == 0 && i % incy == 0)
                                  result[i] = dy[i] - dx[i];
               }

 }