From d7a30ada7e82daf7ce5ca91acac15c1723188fc5 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Mon, 16 Jan 2017 12:51:42 +0300 Subject: [PATCH] Fixed behavior if missing library cudnn.lib --- src/convolutional_kernels.cu | 2 ++ src/convolutional_layer.c | 2 ++ 2 files changed, 4 insertions(+) diff --git a/src/convolutional_kernels.cu b/src/convolutional_kernels.cu index 4191efec..005269b0 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 798d9c16..cf5d2529 100644 --- a/src/convolutional_layer.c +++ b/src/convolutional_layer.c @@ -8,7 +8,9 @@ #include #include +#ifdef CUDNN #pragma comment(lib, "cudnn.lib") +#endif #ifdef AI2 #include "xnor_layer.h"