| 40 | } |
| 41 | |
| 42 | void Timer::Stop() { |
| 43 | if (running()) { |
| 44 | if (Caffe::mode() == Caffe::GPU) { |
| 45 | #ifndef CPU_ONLY |
| 46 | CUDA_CHECK(cudaEventRecord(stop_gpu_, 0)); |
| 47 | #else |
| 48 | NO_GPU; |
| 49 | #endif |
| 50 | } else { |
| 51 | stop_cpu_ = boost::posix_time::microsec_clock::local_time(); |
| 52 | } |
| 53 | running_ = false; |
| 54 | } |
| 55 | } |
| 56 | |
| 57 | |
| 58 | float Timer::MicroSeconds() { |
no outgoing calls