src/cuda.c
@@ -1,9 +1,13 @@ int gpu_index = 0; #ifdef GPU #include "cuda.h" #include "utils.h" #include "blas.h" #include "assert.h" #include <stdlib.h> int gpu_index = 0; void check_error(cudaError_t status) { @@ -12,6 +16,7 @@ const char *s = cudaGetErrorString(status); char buffer[256]; printf("CUDA Error: %s\n", s); assert(0); snprintf(buffer, 256, "CUDA Error: %s", s); error(buffer); } @@ -96,4 +101,4 @@ check_error(status); } #endif