#2390: use timing::getCurrentTime() when trace is enabled

DARMA-tasking · Jan 14, 2025 · ed8151f · ed8151f
1 parent c76a334
commit ed8151f
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/vt/runnable/runnable.cc b/src/vt/runnable/runnable.cc
@@ -124,7 +124,16 @@ void RunnableNew::run() {
   {
     needs_time = contexts_.lb.needsTime();
   }
-  TimeType start_time = needs_time ? theSched()->getRecentTime() : TimeType{NAN};
+
+  auto getTime = [&]() {
+#if vt_check_enabled(trace_enabled)
+  return theTrace() ? timing::getCurrentTime() : theSched()->getRecentTime();
+#else
+  return theSched()->getRecentTime();
+#endif
+};
+
+  TimeType start_time = needs_time ? getTime() : TimeType{NAN};
 
 #if vt_check_enabled(fcontext)
   if (suspended_) {
@@ -175,7 +184,7 @@ void RunnableNew::run() {
 #endif
   }
   theSched()->setRecentTimeToStale();
-  TimeType end_time = needs_time ? theSched()->getRecentTime() : TimeType{NAN};
+  TimeType end_time = needs_time ? getTime() : TimeType{NAN};