src/cuda.h @@ -30,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();