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 @@ -26,6 +30,7 @@ void cuda_random(float *x_gpu, size_t n); float cuda_compare(float *x_gpu, float *x, size_t n, char *s); dim3 cuda_gridsize(size_t n); cudaStream_t get_cuda_stream(); #ifdef CUDNN cudnnHandle_t cudnn_handle();