From 3d30510a50007e03067a028d4585bb1c358d49dc Mon Sep 17 00:00:00 2001
From: philkr <philkr@users.noreply.github.com>
Date: Fri, 6 Mar 2015 16:49:19 -0800
Subject: [PATCH] Making python layer work with cmake

---
 CMakeLists.txt           | 1 +
 cmake/Dependencies.cmake | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 54b044d3..74fa70c9 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -20,6 +20,7 @@ caffe_option(BUILD_python "Build Python wrapper" ON)
 set(python_version "2" CACHE STRING "Specify which python version to use")
 caffe_option(BUILD_matlab "Build Matlab wrapper" OFF IF UNIX OR APPLE)
 caffe_option(BUILD_docs   "Build documentation" ON IF UNIX OR APPLE)
+caffe_option(BUILD_python_layer "Build the caffe python layer" ON)
 
 # ---[ Dependencies
 include(cmake/Dependencies.cmake)
diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake
index b1ac96c6..8453b539 100644
--- a/cmake/Dependencies.cmake
+++ b/cmake/Dependencies.cmake
@@ -127,6 +127,11 @@ if(BUILD_python)
   endif()
   if(PYTHONLIBS_FOUND AND NUMPY_FOUND AND Boost_PYTHON_FOUND)
     set(HAVE_PYTHON TRUE)
+    if(BUILD_python_layer)
+      add_definitions(-DWITH_PYTHON_LAYER)
+      include_directories(SYSTEM ${PYTHON_INCLUDE_DIRS} ${NUMPY_INCLUDE_DIR} ${Boost_INCLUDE_DIRS})
+      list(APPEND Caffe_LINKER_LIBS ${PYTHON_LIBRARIES} ${Boost_LIBRARIES})
+    endif()
   endif()
 endif()