| 108 | |
| 109 | template<typename Dtype> |
| 110 | void GPUParams<Dtype>::configure(Solver<Dtype>* solver) const { |
| 111 | const vector<Blob<Dtype>*>& net = |
| 112 | solver->net()->learnable_params(); |
| 113 | apply_buffers(net, data_, size_, replace_gpu); |
| 114 | apply_buffers(net, diff_, size_, replace_gpu_diff); |
| 115 | } |
| 116 | |
| 117 | void DevicePair::compute(const vector<int> devices, vector<DevicePair>* pairs) { |
| 118 | #ifndef CPU_ONLY |
no test coverage detected