OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed behavior if missing library cudnn.lib

This commit is contained in:
AlexeyAB 2017-01-16 12:51:42 +03:00
parent e83d8e56d5
commit d7a30ada7e
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/convolutional_kernels.cu
View File

@ -2,7 +2,9 @@
#include "curand.h" #include "curand.h"
#include "cublas_v2.h" #include "cublas_v2.h"
#ifdef CUDNN
#pragma comment(lib, "cudnn.lib") #pragma comment(lib, "cudnn.lib")
#endif
extern "C" { extern "C" {
#include "convolutional_layer.h" #include "convolutional_layer.h"

2
src/convolutional_layer.c
View File

@ -8,7 +8,9 @@
#include <stdio.h> #include <stdio.h>
#include <time.h> #include <time.h>
#ifdef CUDNN
#pragma comment(lib, "cudnn.lib") #pragma comment(lib, "cudnn.lib")
#endif
#ifdef AI2 #ifdef AI2
#include "xnor_layer.h" #include "xnor_layer.h"