| 115 | |
| 116 | template<typename Dtype> |
| 117 | NCCL<Dtype>::NCCL(shared_ptr<Solver<Dtype> > solver) |
| 118 | : GPUParams<Dtype>(solver, getDevice()), |
| 119 | comm_(), solver_(solver), barrier_() { |
| 120 | this->Configure(solver.get()); |
| 121 | Init(); |
| 122 | } |
| 123 | |
| 124 | template<typename Dtype> |
| 125 | NCCL<Dtype>::NCCL(shared_ptr<Solver<Dtype> > solver, const string& uid) |