add timer

30283570 · Rick Ho · 2b5672e5 · 30283570
Commit 30283570 authored Dec 17, 2020 by Rick Ho
Show whitespace changes
Inline Side-by-side

Showing with 16 additions and 0 deletions

pytorch/cuda/timer.hh pytorch/cuda/timer.hh +16 -0

No files found.
--- a/pytorch/cuda/timer.hh
+++ b/pytorch/cuda/timer.hh
+#ifndef TIMER_HH
+#define TIMER_HH
+
+#include <chrono>
+
+inline double getDuration(std::chrono::time_point<std::chrono::system_clock> a,
+		                    std::chrono::time_point<std::chrono::system_clock> b) {
+	      return  std::chrono::duration<double>(b - a).count();
+}
+
+#define timestamp(__var__) auto __var__ = std::chrono::system_clock::now();
+
+#include <chrono>
+
+#endif  // TIMER_HH
+