Caffe2 - C++ API
A deep learning, cross platform ML framework
communicator_op_gpu.cc
1 
17 #include "caffe2/core/context_gpu.h"
18 #include "caffe2/core/operator.h"
19 #include "caffe2/operators/no_default_engine_op.h"
20 
21 namespace caffe2 {
22 // Communication operators do not have default engines.
23 REGISTER_CUDA_OPERATOR(CreateCommonWorld, NoDefaultEngineOp<CUDAContext>);
24 REGISTER_CUDA_OPERATOR(CloneCommonWorld, NoDefaultEngineOp<CUDAContext>);
25 REGISTER_CUDA_OPERATOR(Broadcast, NoDefaultEngineOp<CUDAContext>);
26 REGISTER_CUDA_OPERATOR(Reduce, NoDefaultEngineOp<CUDAContext>);
27 REGISTER_CUDA_OPERATOR(Allgather, NoDefaultEngineOp<CUDAContext>);
28 REGISTER_CUDA_OPERATOR(Allreduce, NoDefaultEngineOp<CUDAContext>);
29 REGISTER_CUDA_OPERATOR(SendTensor, NoDefaultEngineOp<CUDAContext>);
30 REGISTER_CUDA_OPERATOR(ReceiveTensor, NoDefaultEngineOp<CUDAContext>);
31 
32 } // namespace caffe2
Copyright (c) 2016-present, Facebook, Inc.