| 40 | } |
| 41 | |
| 42 | void InternalThread::entry(int device, Caffe::Brew mode, int rand_seed, |
| 43 | int solver_count, int solver_rank, bool multiprocess) { |
| 44 | #ifndef CPU_ONLY |
| 45 | CUDA_CHECK(cudaSetDevice(device)); |
| 46 | #endif |
| 47 | Caffe::set_mode(mode); |
| 48 | Caffe::set_random_seed(rand_seed); |
| 49 | Caffe::set_solver_count(solver_count); |
| 50 | Caffe::set_solver_rank(solver_rank); |
| 51 | Caffe::set_multiprocess(multiprocess); |
| 52 | |
| 53 | InternalThreadEntry(); |
| 54 | } |
| 55 | |
| 56 | void InternalThread::StopInternalThread() { |
| 57 | if (is_started()) { |
nothing calls this directly
no test coverage detected