start level for LB

RevathiJambunathan · May 3, 2024 · 5788d74 · 5788d74
1 parent 8c137ca
commit 5788d74
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/Source/Parallelization/WarpXRegrid.cpp b/Source/Parallelization/WarpXRegrid.cpp
@@ -170,8 +170,8 @@ WarpX::LoadBalance ()
         //        ablastr::coarsen::sample::Coarsen(*costs[lev-1], *costs[lev],0,0,1,0,WarpX::RefRatio(lev-1));
         //    }
         //}
-
-        for (int lev = 0; lev <= nLevels; ++lev)
+        int startlev = WarpX::load_balance_startlevel;
+        for (int lev = startlev; lev <= nLevels; ++lev)
         {
             int doLoadBalance = false;
 

diff --git a/Source/WarpX.H b/Source/WarpX.H
@@ -708,7 +708,7 @@ public:
     {
         return load_balance_intervals;
     }
-
+    static int load_balance_startlevel;
     /**
      * \brief Private function for spectral solver
      * Applies a damping factor in the guards cells that extend

diff --git a/Source/WarpX.cpp b/Source/WarpX.cpp
@@ -127,6 +127,7 @@ bool WarpX::do_single_precision_comms = false;
 bool WarpX::do_subcycle_current = false;
 int WarpX::n_subcycle_current = 1;
 bool WarpX::do_abc_in_pml = false;
+int WarpX::load_balance_startlevel = 0;
 
 bool WarpX::do_shared_mem_charge_deposition = false;
 bool WarpX::do_shared_mem_current_deposition = false;
@@ -1353,6 +1354,7 @@ WarpX::ReadParameters ()
         pp_algo.query("load_balance_with_sfc", load_balance_with_sfc);
         pp_algo.query("do_similar_dm_refpatch", do_similar_dm_refpatch);
         pp_algo.query("do_SFC_dm_vectorlevel",do_SFC_dm_vectorlevel);
+	pp_algo.query("load_balance_startlevel",load_balance_startlevel);
         // Knapsack factor only used with non-SFC strategy
         if (!load_balance_with_sfc) {
             pp_algo.query("load_balance_knapsack_factor", load_balance_knapsack_factor);