Add headers to build on OS X

diff --git a/include/caffe/vision_layers.hpp b/include/caffe/vision_layers.hpp
index 74b27cc..a511f92 100644 (file)
--- a/include/caffe/vision_layers.hpp
+++ b/include/caffe/vision_layers.hpp
@@ -12,6 +12,8 @@
 #include "boost/scoped_ptr.hpp"
 #include "hdf5.h"
 
+#include "caffe/blob.hpp"
+#include "caffe/common.hpp"
 #include "caffe/layer.hpp"
 #include "caffe/proto/caffe.pb.h"
 

diff --git a/src/caffe/layers/images_layer.cu b/src/caffe/layers/images_layer.cu
index 7f67283..187be33 100644 (file)
--- a/src/caffe/layers/images_layer.cu
+++ b/src/caffe/layers/images_layer.cu
@@ -1,5 +1,6 @@
 // Copyright 2013 Yangqing Jia
 
+#include <cuda_runtime.h>
 #include <stdint.h>
 #include <leveldb/db.h>
 #include <pthread.h>
@@ -9,6 +10,8 @@
 #include <iostream>  // NOLINT(readability/streams)
 #include <fstream>  // NOLINT(readability/streams)
 
+#include "caffe/blob.hpp"
+#include "caffe/common.hpp"
 #include "caffe/layer.hpp"
 #include "caffe/util/io.hpp"
 #include "caffe/vision_layers.hpp"

diff --git a/src/caffe/layers/window_data_layer.cu b/src/caffe/layers/window_data_layer.cu
index bf33194..3994352 100644 (file)
--- a/src/caffe/layers/window_data_layer.cu
+++ b/src/caffe/layers/window_data_layer.cu
@@ -2,6 +2,7 @@
 //
 // Based on data_layer.cpp by Yangqing Jia.
 
+#include <cuda_runtime.h>
 #include <stdint.h>
 #include <pthread.h>
 
@@ -16,6 +17,8 @@
 #include "opencv2/highgui/highgui.hpp"
 #include "opencv2/imgproc/imgproc.hpp"
 
+#include "caffe/blob.hpp"
+#include "caffe/common.hpp"
 #include "caffe/layer.hpp"
 #include "caffe/util/io.hpp"
 #include "caffe/vision_layers.hpp"