Joseph Redmon
2015-03-08 f047cfff99e00e28c02eb59b6d32386c122f9af6
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