Switched multi-GPU to NCCL
authorCyprien Noel <cyprien.noel@gmail.com>
Fri, 6 Jan 2017 22:55:12 +0000 (14:55 -0800)
committerCyprien Noel <cyprien.noel@gmail.com>
Fri, 6 Jan 2017 22:56:39 +0000 (14:56 -0800)
commit3ba20549b7f49a76cd023d19f781a6891b2c2122
tree3786ec8f8c120bdb29166a31d396274c86820fc1
parent2317fa19d3f5a65cb22adcbd3792ea248996744e
Switched multi-GPU to NCCL
48 files changed:
CMakeLists.txt
Makefile
Makefile.config.example
cmake/Dependencies.cmake
cmake/Modules/FindNCCL.cmake [new file with mode: 0644]
cmake/Summary.cmake
include/caffe/blob.hpp
include/caffe/common.hpp
include/caffe/data_reader.hpp [deleted file]
include/caffe/internal_thread.hpp
include/caffe/layer.hpp
include/caffe/layers/base_data_layer.hpp
include/caffe/layers/data_layer.hpp
include/caffe/layers/hdf5_data_layer.hpp
include/caffe/layers/python_layer.hpp
include/caffe/net.hpp
include/caffe/parallel.hpp
include/caffe/solver.hpp
include/caffe/syncedmem.hpp
include/caffe/util/math_functions.hpp
include/caffe/util/nccl.hpp [new file with mode: 0644]
src/caffe/blob.cpp
src/caffe/common.cpp
src/caffe/data_reader.cpp [deleted file]
src/caffe/internal_thread.cpp
src/caffe/layer.cpp
src/caffe/layers/base_data_layer.cpp
src/caffe/layers/base_data_layer.cu
src/caffe/layers/data_layer.cpp
src/caffe/layers/hdf5_data_layer.cpp
src/caffe/layers/hdf5_data_layer.cu
src/caffe/layers/image_data_layer.cpp
src/caffe/layers/window_data_layer.cpp
src/caffe/net.cpp
src/caffe/parallel.cpp
src/caffe/proto/caffe.proto
src/caffe/solver.cpp
src/caffe/solvers/adagrad_solver.cpp
src/caffe/solvers/nesterov_solver.cpp
src/caffe/solvers/sgd_solver.cpp
src/caffe/syncedmem.cpp
src/caffe/test/test_data_layer.cpp
src/caffe/test/test_gradient_based_solver.cpp
src/caffe/test/test_hdf5data_layer.cpp
src/caffe/util/blocking_queue.cpp
src/caffe/util/db_lmdb.cpp
src/caffe/util/math_functions.cu
tools/caffe.cpp