#431 Add new flag for input LB statistics file.

DARMA-tasking · Feb 5, 2020 · f797578 · f797578
1 parent a4c954c
commit f797578
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/vt/configs/arguments/args.cc b/src/vt/configs/arguments/args.cc
@@ -368,6 +368,8 @@ namespace vt { namespace arguments {
   auto lb_stats      = "Enable load balancing statistics";
   auto lb_stats_dir  = "Load balancing statistics output directory";
   auto lb_stats_file = "Load balancing statistics output file name";
+  auto lb_stats_dir_in  = "Load balancing statistics input directory";
+  auto lb_stats_file_in = "Load balancing statistics input file name";
   auto lbn = "NoLB";
   auto lbi = 1;
   auto lbf = "";
@@ -384,6 +386,10 @@ namespace vt { namespace arguments {
   auto ww = app.add_flag("--vt_lb_stats",        vt_lb_stats,       lb_stats);
   auto wx = app.add_option("--vt_lb_stats_dir",  vt_lb_stats_dir,   lb_stats_dir, lbd);
   auto wy = app.add_option("--vt_lb_stats_file", vt_lb_stats_file,  lb_stats_file,lbs);
+
+  auto xx = app.add_option("--vt_lb_stats_dir_in", vt_lb_stats_dir_in, lb_stats_dir_in, lbd);
+  auto xy = app.add_option("--vt_lb_stats_file_in", vt_lb_stats_file_in, lb_stats_file_in,lbs);
+
   auto debugLB = "Load Balancing";
   s->group(debugLB);
   t->group(debugLB);

diff --git a/src/vt/configs/arguments/args.h b/src/vt/configs/arguments/args.h
@@ -95,6 +95,8 @@ struct ArgConfig {
   static bool vt_lb_stats;
   static std::string vt_lb_stats_dir;
   static std::string vt_lb_stats_file;
+  static std::string vt_lb_stats_dir_in;
+  static std::string vt_lb_stats_file_in;
 
   static bool vt_no_detect_hang;
   static bool vt_print_no_progress;