From f047cfff99e00e28c02eb59b6d32386c122f9af6 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sun, 08 Mar 2015 18:31:12 +0000
Subject: [PATCH] renamed sigmoid to logistic
---
src/cuda.c | 9 +++++++--
1 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/src/cuda.c b/src/cuda.c
index 27153ea..c914290 100644
--- a/src/cuda.c
+++ b/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
--
Gitblit v1.10.0