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"