#2388: set lb stats write event on startup

Author: cwschilly

src/vt/trace/trace.cc

@@ -84,6 +84,7 @@ void Trace::initialize() /*override*/ {
 
   // Register a trace user event to demarcate flushes that occur
   flush_event_ = trace::registerEventCollective("trace_flush");
+  write_stats_event_ = trace::registerEventCollective("write_lb_stats");
 #endif
 }
 

src/vt/trace/trace_lite.h

@@ -295,6 +295,15 @@ struct TraceLite  {
     return traces_.size();
   }
 
+  /**
+   * @brief Get the LB Stats Write event
+   *
+   * @return the LB stats write event
+   */
+  UserEventIDType getWriteStatsEvent() const {
+    return write_stats_event_;
+  }
+
   /**
    * @brief Get the last recorded trace event
    *
@@ -430,20 +439,21 @@ struct TraceLite  {
   EventHoldStackType event_holds_;
   TraceStackType open_events_;
   TraceContainerType traces_;
-  TraceEventIDType cur_event_   = 1;
-  UserEventIDType flush_event_  = no_user_event_id;
-  bool enabled_                 = true;
-  TimeType start_time_          = TimeType{0.0};
-  std::string prog_name_        = "";
-  std::string trace_name_       = "";
-  std::string full_trace_name_  = "";
-  std::string full_sts_name_    = "";
-  std::string full_dir_name_    = "";
-  bool wrote_sts_file_          = false;
-  size_t trace_write_count_     = 0;
-  bool standalone_initalized_   = false;
-  bool trace_enabled_cur_phase_ = true;
-  bool idle_begun_              = false;
+  TraceEventIDType cur_event_         = 1;
+  UserEventIDType write_stats_event_  = no_user_event_id;
+  UserEventIDType flush_event_        = no_user_event_id;
+  bool enabled_                       = true;
+  TimeType start_time_                = TimeType{0.0};
+  std::string prog_name_              = "";
+  std::string trace_name_             = "";
+  std::string full_trace_name_        = "";
+  std::string full_sts_name_          = "";
+  std::string full_dir_name_          = "";
+  bool wrote_sts_file_                = false;
+  size_t trace_write_count_           = 0;
+  bool standalone_initalized_         = false;
+  bool trace_enabled_cur_phase_       = true;
+  bool idle_begun_                    = false;
   std::unique_ptr<vt_gzFile> log_file_;
   std::unordered_map<TraceEventIDType, std::stack<Log*>> incomplete_notes_ = {};
 };

src/vt/vrt/collection/balance/lb_invoke/lb_manager.cc

@@ -667,7 +667,7 @@ void LBManager::commitPhaseStatistics(PhaseType phase) {
   }
 
   #if vt_check_enabled(trace_enabled)
-  auto write_lb_stats_event = theTrace()->registerUserEventColl("write_lb_stats");
+  auto write_lb_stats_event = theTrace()->getWriteStatsEvent();
   theTrace()->addUserEventBracketedBegin(write_lb_stats_event);
   #endif