| 138 | |
| 139 | template<typename Dtype> |
| 140 | void NCCL<Dtype>::Init() { |
| 141 | if (solver_->param().layer_wise_reduce()) { |
| 142 | CUDA_CHECK(cudaStreamCreateWithFlags(&stream_, cudaStreamNonBlocking)); |
| 143 | } |
| 144 | } |
| 145 | |
| 146 | template<typename Dtype> |
| 147 | NCCL<Dtype>::~NCCL() { |
nothing calls this directly
no outgoing calls
no test coverage detected