src/cuda.h
@@ -19,6 +19,9 @@ #include "cudnn.h" #endif #ifdef __cplusplus extern "C" { #endif void check_error(cudaError_t status); cublasHandle_t blas_handle(); float *cuda_make_array(float *x, size_t n); @@ -32,6 +35,9 @@ 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 __cplusplus } #endif #ifdef CUDNN cudnnHandle_t cudnn_handle();
