Merge pull request #4609 from intelfx/BVLC-work-buildsystem
authorEvan Shelhamer <shelhamer@imaginarynumber.net>
Fri, 17 Feb 2017 00:35:51 +0000 (16:35 -0800)
committerGitHub <noreply@github.com>
Fri, 17 Feb 2017 00:35:51 +0000 (16:35 -0800)
[cmake] Improvements to the build system

1  2 
CMakeLists.txt
cmake/ConfigGen.cmake
cmake/Cuda.cmake
cmake/Dependencies.cmake
src/caffe/net.cpp

diff --cc CMakeLists.txt
Simple merge
Simple merge
Simple merge
@@@ -63,17 -82,9 +82,16 @@@ if(NOT HAVE_CUDA
      message(WARNING "-- CUDA is not detected by cmake. Building without it...")
    endif()
  
-   # TODO: remove this not cross platform define in future. Use caffe_config.h instead.
-   add_definitions(-DCPU_ONLY)
+   list(APPEND Caffe_DEFINITIONS PUBLIC -DCPU_ONLY)
  endif()
  
 +if(USE_NCCL)
 +  find_package(NCCL REQUIRED)
 +  include_directories(SYSTEM ${NCCL_INCLUDE_DIR})
 +  list(APPEND Caffe_LINKER_LIBS ${NCCL_LIBRARIES})
 +  add_definitions(-DUSE_NCCL)
 +endif()
 +
  # ---[ OpenCV
  if(USE_OPENCV)
    find_package(OpenCV QUIET COMPONENTS core highgui imgproc imgcodecs)
Simple merge