| 19 | } |
| 20 | |
| 21 | void InternalThread::StartInternalThread() { |
| 22 | CHECK(!is_started()) << "Threads should persist and not be restarted."; |
| 23 | |
| 24 | int device = 0; |
| 25 | #ifndef CPU_ONLY |
| 26 | CUDA_CHECK(cudaGetDevice(&device)); |
| 27 | #endif |
| 28 | Caffe::Brew mode = Caffe::mode(); |
| 29 | int rand_seed = caffe_rng_rand(); |
| 30 | int solver_count = Caffe::solver_count(); |
| 31 | int solver_rank = Caffe::solver_rank(); |
| 32 | bool multiprocess = Caffe::multiprocess(); |
| 33 | |
| 34 | try { |
| 35 | thread_.reset(new boost::thread(&InternalThread::entry, this, device, mode, |
| 36 | rand_seed, solver_count, solver_rank, multiprocess)); |
| 37 | } catch (std::exception& e) { |
| 38 | LOG(FATAL) << "Thread exception: " << e.what(); |
| 39 | } |
| 40 | } |
| 41 | |
| 42 | void InternalThread::entry(int device, Caffe::Brew mode, int rand_seed, |
| 43 | int solver_count, int solver_rank, bool multiprocess) { |