Fixed behavior if missing library cudnn.lib

8 years ago · d7a30ada7e
parent e83d8e56d5
commit d7a30ada7e
2 changed files with 4 additions and 0 deletions
--- 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"
--- 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"