From d7a30ada7e82daf7ce5ca91acac15c1723188fc5 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 16 Jan 2017 09:51:42 +0000
Subject: [PATCH] Fixed behavior if missing library cudnn.lib
---
src/convolutional_layer.c | 2 ++
src/convolutional_kernels.cu | 2 ++
2 files changed, 4 insertions(+), 0 deletions(-)
diff --git a/src/convolutional_kernels.cu b/src/convolutional_kernels.cu
index 4191efe..005269b 100644
--- a/src/convolutional_kernels.cu
+++ b/src/convolutional_kernels.cu
@@ -2,7 +2,9 @@
#include "curand.h"
#include "cublas_v2.h"
+#ifdef CUDNN
#pragma comment(lib, "cudnn.lib")
+#endif
extern "C" {
#include "convolutional_layer.h"
diff --git a/src/convolutional_layer.c b/src/convolutional_layer.c
index 798d9c1..cf5d252 100644
--- a/src/convolutional_layer.c
+++ b/src/convolutional_layer.c
@@ -8,7 +8,9 @@
#include <stdio.h>
#include <time.h>
+#ifdef CUDNN
#pragma comment(lib, "cudnn.lib")
+#endif
#ifdef AI2
#include "xnor_layer.h"
--
Gitblit v1.10.0