Add Timer to wrap CPU clock_t and GPU cudaEvent_t based timing

1 files changed, 53 insertions, 0 deletions

diff --git a/src/caffe/util/benchmark.cpp b/src/caffe/util/benchmark.cpp
new file mode 100644
index 00000000..19a036d9
--- /dev/null
+++ b/src/caffe/util/benchmark.cpp
@@ -0,0 +1,53 @@
+// Copyright 2014 kloud@github
+
+#include <ctime>
+#include <cuda_runtime.h>
+
+#include "caffe/common.hpp"
+#include "caffe/util/benchmark.hpp"
+
+namespace caffe {
+
+Timer::Timer() {
+  if (Caffe::mode() == Caffe::GPU) {
+    cudaEventCreate (&start_gpu_);
+    cudaEventCreate (&stop_gpu_);
+  }
+}
+
+Timer::~Timer() {
+  if (Caffe::mode() == Caffe::GPU) {
+    cudaEventDestroy (start_gpu_);
+    cudaEventDestroy (stop_gpu_);
+  }
+}
+
+void Timer::Start() {
+  if (Caffe::mode() == Caffe::GPU) {
+    cudaEventRecord(start_gpu_, 0);
+  } else {
+    start_cpu_ = clock();
+  }
+}
+
+void Timer::Stop() {
+  if (Caffe::mode() == Caffe::GPU) {
+    cudaEventRecord(stop_gpu_, 0);
+  } else {
+    stop_cpu_ = clock();
+  }
+}
+
+float Timer::ElapsedSeconds() {
+  float elapsed;
+  if (Caffe::mode() == Caffe::GPU) {
+    cudaEventSynchronize(stop_gpu_);
+    cudaEventElapsedTime(&elapsed, start_gpu_, stop_gpu_);
+    elapsed /= 1000.;
+  } else {
+    elapsed = float(stop_cpu_ - start_cpu_) / CLOCKS_PER_SEC;
+  }
+  return elapsed;
+}
+
+}  // namespace caffe