blob: 21c38ad36fee0b876483ea49d9b33de3ae87b268 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
// Copyright 2014 kloud@github
#include <boost/date_time/posix_time/posix_time.hpp>
#include <cuda_runtime.h>
#include "caffe/common.hpp"
#include "caffe/util/benchmark.hpp"
namespace caffe {
Timer::Timer()
: initted_(false),
running_(false),
has_run_at_least_once_(false) {
Init();
}
Timer::~Timer() {
if (Caffe::mode() == Caffe::GPU) {
CUDA_CHECK(cudaEventDestroy(start_gpu_));
CUDA_CHECK(cudaEventDestroy(stop_gpu_));
}
}
void Timer::Start() {
if (!running()) {
if (Caffe::mode() == Caffe::GPU) {
CUDA_CHECK(cudaEventRecord(start_gpu_, 0));
} else {
start_cpu_ = boost::posix_time::microsec_clock::local_time();
}
running_ = true;
has_run_at_least_once_ = true;
}
}
void Timer::Stop() {
if (running()) {
if (Caffe::mode() == Caffe::GPU) {
CUDA_CHECK(cudaEventRecord(stop_gpu_, 0));
CUDA_CHECK(cudaEventSynchronize(stop_gpu_));
} else {
stop_cpu_ = boost::posix_time::microsec_clock::local_time();
}
running_ = false;
}
}
float Timer::MilliSeconds() {
if (!has_run_at_least_once()) {
LOG(WARNING) << "Timer has never been run before reading time.";
return 0;
}
if (running()) {
Stop();
}
if (Caffe::mode() == Caffe::GPU) {
CUDA_CHECK(cudaEventElapsedTime(&elapsed_milliseconds_, start_gpu_,
stop_gpu_));
} else {
elapsed_milliseconds_ = (stop_cpu_ - start_cpu_).total_milliseconds();
}
return elapsed_milliseconds_;
}
float Timer::Seconds() {
return MilliSeconds() / 1000.;
}
void Timer::Init() {
if (!initted()) {
if (Caffe::mode() == Caffe::GPU) {
CUDA_CHECK(cudaEventCreate(&start_gpu_));
CUDA_CHECK(cudaEventCreate(&stop_gpu_));
}
initted_ = true;
}
}
} // namespace caffe
|
