1 #ifndef CAFFE_UTIL_NCCL_H_
2 #define CAFFE_UTIL_NCCL_H_
7 #include "caffe/common.hpp"
9 #define NCCL_CHECK(condition) \
11 ncclResult_t result = condition; \
12 CHECK_EQ(result, ncclSuccess) << " " \
13 << ncclGetErrorString(result); \
20 template <
typename Dtype>
class dataType;
22 template<>
class dataType<float> {
24 static const ncclDataType_t type = ncclFloat;
26 template<>
class dataType<double> {
28 static const ncclDataType_t type = ncclDouble;
35 #endif // end USE_NCCL
37 #endif // CAFFE_UTIL_NCCL_H_
A layer factory that allows one to register layers. During runtime, registered layers can be called b...
Definition: blob.hpp:14