src/cuda.h
@@ -1,6 +1,10 @@ #ifndef CUDA_H #define CUDA_H #if defined(_MSC_VER) && _MSC_VER < 1900 #define inline __inline #endif extern int gpu_index; #ifdef GPU @@ -21,6 +25,7 @@ int *cuda_make_int_array(size_t n); void cuda_push_array(float *x_gpu, float *x, size_t n); void cuda_pull_array(float *x_gpu, float *x, size_t n); void cuda_set_device(int n); void cuda_free(float *x_gpu); void cuda_random(float *x_gpu, size_t n); float cuda_compare(float *x_gpu, float *x, size_t n, char *s);