#1899: runnable: add missing getter for trace context

DARMA-tasking · Sep 20, 2022 · 5416018 · 5416018
1 parent aa7d058
commit 5416018
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/vt/runnable/runnable.impl.h b/src/vt/runnable/runnable.impl.h
@@ -63,6 +63,17 @@ inline ctx::LBData* RunnableNew::get<ctx::LBData>() {
     return nullptr;
 }
 
+#if vt_check_enabled(trace_enabled)
+template <>
+inline ctx::Trace* RunnableNew::get<ctx::Trace>() {
+  if (contexts_.has_trace)
+    return &contexts_.trace;
+  else
+    return nullptr;
+}
+#endif
+
+
 template <typename... Args>
 void RunnableNew::addContextSetContext(Args&&... args) {
   contexts_.setcontext = ctx::SetContext{std::forward<Args>(args)...};