ff33c46016b8928c1609cf905308577aecffa4a8,scikits/cuda/misc.py,,,#,477

Before Change

    return y_gpu

if __name__ == "__main__":
    import doctest
    doctest.testmod()

After Change

    return c_gpu
    
diff_mod_template = Template(
//include <pycuda/pycuda-complex.hpp>

//define USE_DOUBLE ${use_double}
//define USE_COMPLEX ${use_complex}

//if USE_DOUBLE == 1
//define REAL_TYPE double
//if USE_COMPLEX == 1
//define TYPE pycuda::complex<double>
//else
//define TYPE double
//endif
//else
//define REAL_TYPE float
//if USE_COMPLEX == 1
//define TYPE pycuda::complex<float>
//else
//define TYPE float
//endif
//endif

__global__ void diff(TYPE *x, TYPE *y, unsigned int N) {
    unsigned int idx = blockIdx.y*${max_threads_per_block}*${max_blocks_per_grid}+
                       blockIdx.x*${max_threads_per_block}+threadIdx.x;

    if (idx < N-1) {
        y[idx] = x[idx+1]-x[idx];
    }
}
)

def diff(x_gpu):
    
    Calculate the discrete difference.

Instances