diff --git a/src/vt/runnable/runnable.cc b/src/vt/runnable/runnable.cc index 15ba22af36..b7720c6871 100644 --- a/src/vt/runnable/runnable.cc +++ b/src/vt/runnable/runnable.cc @@ -212,33 +212,33 @@ void RunnableNew::run() { #endif } -void RunnableNew::begin() { +void RunnableNew::begin(TimeType current_time) { contexts_.setcontext.begin(); if (contexts_.has_td) contexts_.td.begin(); if (contexts_.has_col) contexts_.col.begin(); - if (contexts_.has_lb) contexts_.lb.begin(); + if (contexts_.has_lb) contexts_.lb.begin(current_time); #if vt_check_enabled(trace_enabled) if (contexts_.has_trace) contexts_.trace.begin(); #endif } -void RunnableNew::end() { +void RunnableNew::end(TimeType current_time) { contexts_.setcontext.end(); if (contexts_.has_td) contexts_.td.end(); if (contexts_.has_col) contexts_.col.end(); if (contexts_.has_cont) contexts_.cont.end(); - if (contexts_.has_lb) contexts_.lb.end(); + if (contexts_.has_lb) contexts_.lb.end(current_time); #if vt_check_enabled(trace_enabled) if (contexts_.has_trace) contexts_.trace.end(); #endif } -void RunnableNew::suspend() { +void RunnableNew::suspend(TimeType current_time) { #if vt_check_enabled(fcontext) contexts_.setcontext.suspend(); if (contexts_.has_td) contexts_.td.suspend(); if (contexts_.has_col) contexts_.col.suspend(); - if (contexts_.has_lb) contexts_.lb.suspend(); + if (contexts_.has_lb) contexts_.lb.suspend(current_time); # if vt_check_enabled(trace_enabled) if (contexts_.has_trace) contexts_.trace.suspend(); @@ -246,12 +246,12 @@ void RunnableNew::suspend() { #endif } -void RunnableNew::resume() { +void RunnableNew::resume(TimeType current_time) { #if vt_check_enabled(fcontext) contexts_.setcontext.resume(); if (contexts_.has_td) contexts_.td.resume(); if (contexts_.has_col) contexts_.col.resume(); - if (contexts_.has_lb) contexts_.lb.resume(); + if (contexts_.has_lb) contexts_.lb.resume(current_time); # if vt_check_enabled(trace_enabled) if (contexts_.has_trace) contexts_.trace.resume(); diff --git a/src/vt/runnable/runnable.h b/src/vt/runnable/runnable.h index 8691457262..d049da9707 100644 --- a/src/vt/runnable/runnable.h +++ b/src/vt/runnable/runnable.h @@ -236,25 +236,25 @@ struct RunnableNew { * \internal \brief Loop through all the contexts associated with this * runnable and invoke \c begin() on them. */ - void begin(); + void begin(TimeType current_time = 0.0); /** * \internal \brief Loop through all the contexts associated with this * runnable and invoke \c end() on them. */ - void end(); + void end(TimeType current_time = 0.0); /** * \internal \brief Loop through all the contexts associated with this * runnable and invoke \c suspend() on them. */ - void suspend(); + void suspend(TimeType current_time = 0.0); /** * \internal \brief Loop through all the contexts associated with this * runnable and invoke \c resume() on them. */ - void resume(); + void resume(TimeType current_time = 0.0); public: /**