Fixed compile error

pull/492/head
AlexeyAB 7 years ago
parent e947bf29af
commit 2f52cfeb07
  1. 2
      src/yolo_v2_class.cpp

@ -22,12 +22,14 @@ extern "C" {
#define FRAMES 3 #define FRAMES 3
#ifdef GPU
void check_cuda(cudaError_t status) { void check_cuda(cudaError_t status) {
if (status != cudaSuccess) { if (status != cudaSuccess) {
const char *s = cudaGetErrorString(status); const char *s = cudaGetErrorString(status);
printf("CUDA Error Prev: %s\n", s); printf("CUDA Error Prev: %s\n", s);
} }
} }
#endif
struct detector_gpu_t { struct detector_gpu_t {
float **probs; float **probs;

Loading…
Cancel
Save