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