| 80 | } |
| 81 | |
| 82 | float Timer::MilliSeconds() { |
| 83 | if (!has_run_at_least_once()) { |
| 84 | LOG(WARNING) << "Timer has never been run before reading time."; |
| 85 | return 0; |
| 86 | } |
| 87 | if (running()) { |
| 88 | Stop(); |
| 89 | } |
| 90 | if (Caffe::mode() == Caffe::GPU) { |
| 91 | #ifndef CPU_ONLY |
| 92 | CUDA_CHECK(cudaEventElapsedTime(&elapsed_milliseconds_, start_gpu_, |
| 93 | stop_gpu_)); |
| 94 | #else |
| 95 | NO_GPU; |
| 96 | #endif |
| 97 | } else { |
| 98 | elapsed_milliseconds_ = (stop_cpu_ - start_cpu_).total_milliseconds(); |
| 99 | } |
| 100 | return elapsed_milliseconds_; |
| 101 | } |
| 102 | |
| 103 | float Timer::Seconds() { |
| 104 | return MilliSeconds() / 1000.; |
no outgoing calls