-
Notifications
You must be signed in to change notification settings - Fork 18.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #1112 from BVLC/next
Next: release candidate
- Loading branch information
Showing
297 changed files
with
17,723 additions
and
3,255 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,90 @@ | ||
| cmake_minimum_required(VERSION 2.8.8) | ||
| project( Caffe ) | ||
|
|
||
| ### Build Options ########################################################################## | ||
|
|
||
| option(CPU_ONLY "Build Caffe without GPU support" OFF) | ||
| option(BUILD_PYTHON "Build Python wrapper" OFF) | ||
| option(BUILD_MATLAB "Build Matlab wrapper" OFF) | ||
| option(BUILD_EXAMPLES "Build examples" ON) | ||
| option(BUILD_SHARED_LIBS "Build SHARED libs if ON and STATIC otherwise" OFF) | ||
|
|
||
| if(NOT BLAS) | ||
| set(BLAS atlas) | ||
| endif() | ||
|
|
||
| if(NOT CUDA_TEST_DEVICE) | ||
| set(CUDA_TEST_DEVICE -1) | ||
| endif() | ||
|
|
||
| # Install Prefix | ||
| if (CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT) | ||
| set (CMAKE_INSTALL_PREFIX "${CMAKE_BINARY_DIR}/install" CACHE PATH "Default install path" FORCE ) | ||
| endif() | ||
|
|
||
| ### Configuration ########################################################################### | ||
| # Compiler Flags | ||
| set(CMAKE_CXX_COMPILER_FLAGS ${CMAKE_CXX_COMPILER_FLAGS} -Wall) | ||
| set(CMAKE_CXX_FLAGS ${CMAKE_CXX_FLAGS} -fPIC) # set global flags | ||
| set(CMAKE_CXX_FLAGS_DEBUG ${CMAKE_CXX_FLAGS_DEBUG}) # set debug flags | ||
| set(CMAKE_CXX_FLAGS_RELEASE ${CMAKE_CXX_FLAGS_RELEASE}) # set release flags | ||
|
|
||
| # Global Definitions | ||
| if(CPU_ONLY) | ||
| add_definitions(-DCPU_ONLY) | ||
| endif() | ||
|
|
||
| # Include Directories | ||
| set(${PROJECT_NAME}_INCLUDE_DIRS ${CMAKE_SOURCE_DIR}/include) | ||
| include_directories(${${PROJECT_NAME}_INCLUDE_DIRS}) | ||
| include_directories(${CMAKE_SOURCE_DIR}/src) | ||
|
|
||
| # CMake Scripts dir | ||
| set(CMAKE_SCRIPT_DIR ${CMAKE_SOURCE_DIR}/CMakeScripts) | ||
|
|
||
| # CMake module path for custom module finding | ||
| set( CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_SCRIPT_DIR}) | ||
|
|
||
| # CUDA is required globally | ||
| if(NOT CPU_ONLY) | ||
| find_package(CUDA 5.5 REQUIRED) | ||
| include_directories(${CUDA_INCLUDE_DIRS}) | ||
| endif() | ||
|
|
||
| ### Subdirectories ########################################################################## | ||
|
|
||
| add_subdirectory(src/gtest) | ||
| add_subdirectory(src/caffe) | ||
| add_subdirectory(tools) | ||
|
|
||
| if(BUILD_EXAMPLES) | ||
| message(STATUS "Examples enabled") | ||
| add_subdirectory(examples) | ||
| endif() | ||
|
|
||
| if(BUILD_PYTHON) | ||
| message(STATUS "Python enabled") | ||
| add_subdirectory(python) | ||
| endif() | ||
|
|
||
| if(BUILD_MATLAB) | ||
| message(STATUS "Matlab enabled") | ||
| add_subdirectory(matlab) | ||
| endif() | ||
|
|
||
| ### Lint Target Setup ########################################################################## | ||
|
|
||
| set(LINT_TARGET lint) | ||
| set(LINT_SCRIPT ${CMAKE_SCRIPT_DIR}/lint.cmake) | ||
| add_custom_target( | ||
| ${LINT_TARGET} | ||
| COMMAND ${CMAKE_COMMAND} -P ${LINT_SCRIPT} | ||
| ) | ||
|
|
||
| ### Install ################################################################################# | ||
|
|
||
| # Install Includes | ||
| file(GLOB folders ${${PROJECT_NAME}_INCLUDE_DIRS}/*) | ||
| install(DIRECTORY ${folders} DESTINATION include) | ||
|
|
||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,61 @@ | ||
| # Find the Atlas (and Lapack) libraries | ||
| # | ||
| # The following variables are optionally searched for defaults | ||
| # Atlas_ROOT_DIR: Base directory where all Atlas components are found | ||
| # | ||
| # The following are set after configuration is done: | ||
| # Atlas_FOUND | ||
| # Atlas_INCLUDE_DIRS | ||
| # Atlas_LIBRARIES | ||
| # Atlas_LIBRARYRARY_DIRS | ||
|
|
||
| set(Atlas_INCLUDE_SEARCH_PATHS | ||
| /usr/include/atlas | ||
| /usr/include/atlas-base | ||
| $ENV{Atlas_ROOT_DIR} | ||
| $ENV{Atlas_ROOT_DIR}/include | ||
| ) | ||
|
|
||
| set(Atlas_LIB_SEARCH_PATHS | ||
| /usr/lib/atlas | ||
| /usr/lib/atlas-base | ||
| $ENV{Atlas_ROOT_DIR} | ||
| $ENV{Atlas_ROOT_DIR}/lib | ||
| ) | ||
|
|
||
| find_path(Atlas_CBLAS_INCLUDE_DIR NAMES cblas.h PATHS ${Atlas_INCLUDE_SEARCH_PATHS}) | ||
| find_path(Atlas_CLAPACK_INCLUDE_DIR NAMES clapack.h PATHS ${Atlas_INCLUDE_SEARCH_PATHS}) | ||
| find_library(Atlas_CBLAS_LIBRARY NAMES ptcblas_r ptcblas cblas_r cblas PATHS ${Atlas_LIB_SEARCH_PATHS}) | ||
| find_library(Atlas_BLAS_LIBRARY NAMES atlas_r atlas PATHS ${Atlas_LIB_SEARCH_PATHS}) | ||
| find_library(Atlas_LAPACK_LIBRARY NAMES alapack_r alapack lapack_atlas PATHS ${Atlas_LIB_SEARCH_PATHS}) | ||
|
|
||
| set(LOOKED_FOR | ||
|
|
||
| Atlas_CBLAS_INCLUDE_DIR | ||
| Atlas_CLAPACK_INCLUDE_DIR | ||
|
|
||
| Atlas_CBLAS_LIBRARY | ||
| Atlas_BLAS_LIBRARY | ||
| Atlas_LAPACK_LIBRARY | ||
| ) | ||
|
|
||
| include(FindPackageHandleStandardArgs) | ||
| find_package_handle_standard_args(Atlas DEFAULT_MSG ${LOOKED_FOR}) | ||
|
|
||
| if(ATLAS_FOUND) | ||
|
|
||
| mark_as_advanced(${LOOKED_FOR}) | ||
|
|
||
| set(Atlas_INCLUDE_DIR | ||
| ${Atlas_CBLAS_INCLUDE_DIR} | ||
| ${Atlas_CLAPACK_INCLUDE_DIR} | ||
| ) | ||
|
|
||
| set(Atlas_LIBRARIES | ||
| ${Atlas_LAPACK_LIBRARY} | ||
| ${Atlas_CBLAS_LIBRARY} | ||
| ${Atlas_BLAS_LIBRARY} | ||
| ) | ||
|
|
||
| endif(ATLAS_FOUND) | ||
|
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,48 @@ | ||
| # - Try to find GFLAGS | ||
| # | ||
| # The following variables are optionally searched for defaults | ||
| # GFLAGS_ROOT_DIR: Base directory where all GFLAGS components are found | ||
| # | ||
| # The following are set after configuration is done: | ||
| # GFLAGS_FOUND | ||
| # GFLAGS_INCLUDE_DIRS | ||
| # GFLAGS_LIBRARIES | ||
| # GFLAGS_LIBRARYRARY_DIRS | ||
|
|
||
| include(FindPackageHandleStandardArgs) | ||
|
|
||
| set(GFLAGS_ROOT_DIR "" CACHE PATH "Folder contains Gflags") | ||
|
|
||
| # We are testing only a couple of files in the include directories | ||
| if(WIN32) | ||
| find_path(GFLAGS_INCLUDE_DIR gflags/gflags.h | ||
| PATHS ${GFLAGS_ROOT_DIR}/src/windows) | ||
| else() | ||
| find_path(GFLAGS_INCLUDE_DIR gflags/gflags.h | ||
| PATHS ${GFLAGS_ROOT_DIR}) | ||
| endif() | ||
|
|
||
| if(MSVC) | ||
| find_library(GFLAGS_LIBRARY_RELEASE | ||
| NAMES libgflags | ||
| PATHS ${GFLAGS_ROOT_DIR} | ||
| PATH_SUFFIXES Release) | ||
|
|
||
| find_library(GFLAGS_LIBRARY_DEBUG | ||
| NAMES libgflags-debug | ||
| PATHS ${GFLAGS_ROOT_DIR} | ||
| PATH_SUFFIXES Debug) | ||
|
|
||
| set(GFLAGS_LIBRARY optimized ${GFLAGS_LIBRARY_RELEASE} debug ${GFLAGS_LIBRARY_DEBUG}) | ||
| else() | ||
| find_library(GFLAGS_LIBRARY gflags) | ||
| endif() | ||
|
|
||
| find_package_handle_standard_args(GFLAGS DEFAULT_MSG | ||
| GFLAGS_INCLUDE_DIR GFLAGS_LIBRARY) | ||
|
|
||
|
|
||
| if(GFLAGS_FOUND) | ||
| set(GFLAGS_INCLUDE_DIRS ${GFLAGS_INCLUDE_DIR}) | ||
| set(GFLAGS_LIBRARIES ${GFLAGS_LIBRARY}) | ||
| endif() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,48 @@ | ||
| # - Try to find Glog | ||
| # | ||
| # The following variables are optionally searched for defaults | ||
| # GLOG_ROOT_DIR: Base directory where all GLOG components are found | ||
| # | ||
| # The following are set after configuration is done: | ||
| # GLOG_FOUND | ||
| # GLOG_INCLUDE_DIRS | ||
| # GLOG_LIBRARIES | ||
| # GLOG_LIBRARYRARY_DIRS | ||
|
|
||
| include(FindPackageHandleStandardArgs) | ||
|
|
||
| set(GLOG_ROOT_DIR "" CACHE PATH "Folder contains Google glog") | ||
|
|
||
| if(WIN32) | ||
| find_path(GLOG_INCLUDE_DIR glog/logging.h | ||
| PATHS ${GLOG_ROOT_DIR}/src/windows) | ||
| else() | ||
| find_path(GLOG_INCLUDE_DIR glog/logging.h | ||
| PATHS ${GLOG_ROOT_DIR}) | ||
| endif() | ||
|
|
||
| if(MSVC) | ||
| find_library(GLOG_LIBRARY_RELEASE libglog_static | ||
| PATHS ${GLOG_ROOT_DIR} | ||
| PATH_SUFFIXES Release) | ||
|
|
||
| find_library(GLOG_LIBRARY_DEBUG libglog_static | ||
| PATHS ${GLOG_ROOT_DIR} | ||
| PATH_SUFFIXES Debug) | ||
|
|
||
| set(GLOG_LIBRARY optimized ${GLOG_LIBRARY_RELEASE} debug ${GLOG_LIBRARY_DEBUG}) | ||
| else() | ||
| find_library(GLOG_LIBRARY glog | ||
| PATHS ${GLOG_ROOT_DIR} | ||
| PATH_SUFFIXES | ||
| lib | ||
| lib64) | ||
| endif() | ||
|
|
||
| find_package_handle_standard_args(GLOG DEFAULT_MSG | ||
| GLOG_INCLUDE_DIR GLOG_LIBRARY) | ||
|
|
||
| if(GLOG_FOUND) | ||
| set(GLOG_INCLUDE_DIRS ${GLOG_INCLUDE_DIR}) | ||
| set(GLOG_LIBRARIES ${GLOG_LIBRARY}) | ||
| endif() |
Oops, something went wrong.