CMakeLists.txt

   1 cmake_minimum_required(VERSION 2.8.7)
   2 if(POLICY CMP0046)
   3   cmake_policy(SET CMP0046 NEW)
   4 endif()
   5 if(POLICY CMP0054)
   6   cmake_policy(SET CMP0054 NEW)
   7 endif()
   8
   9 # ---[ Caffe project
  10 project(Caffe C CXX)
  11
  12 # ---[ Using cmake scripts and modules
  13 list(APPEND CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR}/cmake/Modules)
  14
  15 include(ExternalProject)
  16
  17 include(cmake/Utils.cmake)
  18 include(cmake/Targets.cmake)
  19 include(cmake/Misc.cmake)
  20 include(cmake/Summary.cmake)
  21 include(cmake/ConfigGen.cmake)
  22
  23 # ---[ Options
  24 caffe_option(CPU_ONLY  "Build Caffe without CUDA support" OFF) # TODO: rename to USE_CUDA
  25 caffe_option(USE_CUDNN "Build Caffe with cuDNN library support" ON IF NOT CPU_ONLY)
  26 caffe_option(BUILD_SHARED_LIBS "Build shared libraries" ON)
  27 caffe_option(BUILD_python "Build Python wrapper" ON)
  28 set(python_version "2" CACHE STRING "Specify which Python version to use")
  29 caffe_option(BUILD_matlab "Build Matlab wrapper" OFF IF UNIX OR APPLE)
  30 caffe_option(BUILD_docs   "Build documentation" ON IF UNIX OR APPLE)
  31 caffe_option(BUILD_python_layer "Build the Caffe Python layer" ON)
  32 caffe_option(USE_OPENCV "Build with OpenCV support" ON)
  33 caffe_option(USE_LEVELDB "Build with levelDB" ON)
  34 caffe_option(USE_LMDB "Build with lmdb" ON)
  35 caffe_option(ALLOW_LMDB_NOLOCK "Allow MDB_NOLOCK when reading LMDB files (only if necessary)" OFF)
  36
  37 # ---[ Dependencies
  38 include(cmake/Dependencies.cmake)
  39
  40 # ---[ Flags
  41 if(UNIX OR APPLE)
  42   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC -Wall")
  43 endif()
  44
  45 if(USE_libstdcpp)
  46   set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -stdlib=libstdc++")
  47   message("-- Warning: forcing libstdc++ (controlled by USE_libstdcpp option in cmake)")
  48 endif()
  49
  50 add_definitions(-DGTEST_USE_OWN_TR1_TUPLE)
  51
  52 # ---[ Warnings
  53 caffe_warnings_disable(CMAKE_CXX_FLAGS -Wno-sign-compare -Wno-uninitialized)
  54
  55 # ---[ Config generation
  56 configure_file(cmake/Templates/caffe_config.h.in "${PROJECT_BINARY_DIR}/caffe_config.h")
  57
  58 # ---[ Includes
  59 set(Caffe_INCLUDE_DIR ${PROJECT_SOURCE_DIR}/include)
  60 include_directories(${Caffe_INCLUDE_DIR} ${PROJECT_BINARY_DIR})
  61 include_directories(BEFORE src) # This is needed for gtest.
  62
  63 # ---[ Subdirectories
  64 add_subdirectory(src/gtest)
  65 add_subdirectory(src/caffe)
  66 add_subdirectory(tools)
  67 add_subdirectory(examples)
  68 add_subdirectory(python)
  69 add_subdirectory(matlab)
  70 add_subdirectory(docs)
  71
  72 # ---[ Linter target
  73 add_custom_target(lint COMMAND ${CMAKE_COMMAND} -P ${PROJECT_SOURCE_DIR}/cmake/lint.cmake)
  74
  75 # ---[ pytest target
  76 if(BUILD_python)
  77   add_custom_target(pytest COMMAND python${python_version} -m unittest discover -s caffe/test WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}/python )
  78   add_dependencies(pytest pycaffe)
  79 endif()
  80
  81 # ---[ Configuration summary
  82 caffe_print_configuration_summary()
  83
  84 # ---[ Export configs generation
  85 caffe_generate_export_configs()