Compile fix

XHuaSheng · Apr 5, 2018 · f1ad9a0 · f1ad9a0
1 parent 943f6e8
commit f1ad9a0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/network_kernels.cu b/src/network_kernels.cu
@@ -36,10 +36,13 @@ extern "C" {
 #include "blas.h"
 }
 
+#ifdef OPENCV
+#include "opencv2/highgui/highgui_c.h"
+#endif
+
 float * get_network_output_gpu_layer(network net, int i);
 float * get_network_delta_gpu_layer(network net, int i);
 float * get_network_output_gpu(network net);
-#include "opencv2/highgui/highgui_c.h"
 
 void forward_network_gpu(network net, network_state state)
 {