From 2f62fe33c913cd9484fe7f2486889d12292c66e0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 07 Feb 2015 02:53:53 +0000
Subject: [PATCH] saving weight files as binaries, hell yeah
---
src/cuda.c | 7 +++++--
1 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/src/cuda.c b/src/cuda.c
index 27153ea..8849fb1 100644
--- a/src/cuda.c
+++ b/src/cuda.c
@@ -1,9 +1,12 @@
+int gpu_index = 0;
+
+#ifdef GPU
+
#include "cuda.h"
#include "utils.h"
#include "blas.h"
#include <stdlib.h>
-int gpu_index = 0;
void check_error(cudaError_t status)
{
@@ -96,4 +99,4 @@
check_error(status);
}
-
+#endif
--
Gitblit v1.10.0