AlexeyAB
2017-01-16 d7a30ada7e82daf7ce5ca91acac15c1723188fc5
Fixed behavior if missing library cudnn.lib
2 files modified
4 ■■■■ changed files
src/convolutional_kernels.cu 2 ●●●●● patch | view | raw | blame | history
src/convolutional_layer.c 2 ●●●●● patch | view | raw | blame | history
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"
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"