From 979d02126b1a597361934f86f50eeda31ff083fe Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Mon, 09 Feb 2015 21:27:58 +0000
Subject: [PATCH] Generalizing conv layer so deconv is easier
---
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