1 cmake_minimum_required(VERSION 2.8.7)
6 # ---[ Using cmake scripts and modules
7 list(APPEND CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR}/cmake/Modules)
9 include(ExternalProject)
11 include(cmake/Utils.cmake)
12 include(cmake/Targets.cmake)
13 include(cmake/Misc.cmake)
14 include(cmake/Summary.cmake)
15 include(cmake/ConfigGen.cmake)
18 caffe_option(CPU_ONLY "Build Caffe wihtout CUDA support" OFF) # TODO: rename to USE_CUDA
19 caffe_option(USE_CUDNN "Build Caffe with cuDNN libary support" ON IF NOT CPU_ONLY)
20 caffe_option(BUILD_SHARED_LIBS "Build shared libraries" ON)
21 caffe_option(BUILD_python "Build Python wrapper" ON)
22 set(python_version "2" CACHE STRING "Specify which python version to use")
23 caffe_option(BUILD_matlab "Build Matlab wrapper" OFF IF UNIX OR APPLE)
24 caffe_option(BUILD_docs "Build documentation" ON IF UNIX OR APPLE)
25 caffe_option(BUILD_python_layer "Build the caffe python layer" ON)
28 include(cmake/Dependencies.cmake)
32 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -fPIC -Wall")
36 set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -stdlib=libstdc++")
37 message("-- Warning: forcing libstdc++ (controlled by USE_libstdcpp option in cmake)")
40 add_definitions(-DGTEST_USE_OWN_TR1_TUPLE)
43 caffe_warnings_disable(CMAKE_CXX_FLAGS -Wno-sign-compare -Wno-uninitialized)
45 # ---[ Config generation
46 configure_file(cmake/Templates/caffe_config.h.in "${PROJECT_BINARY_DIR}/caffe_config.h")
49 set(Caffe_INCLUDE_DIR ${PROJECT_SOURCE_DIR}/include)
50 include_directories(${Caffe_INCLUDE_DIR} ${PROJECT_BINARY_DIR})
51 include_directories(BEFORE src) # This is needed for gtest.
54 add_subdirectory(src/gtest)
55 add_subdirectory(src/caffe)
56 add_subdirectory(tools)
57 add_subdirectory(examples)
58 add_subdirectory(python)
59 add_subdirectory(matlab)
60 add_subdirectory(docs)
63 add_custom_target(lint COMMAND ${CMAKE_COMMAND} -P ${PROJECT_SOURCE_DIR}/cmake/lint.cmake)
65 # ---[ Configuration summary
66 caffe_print_configuration_summary()
68 # ---[ Export configs generation
69 caffe_generate_export_configs()