Only rank0 writes energy timings

parthenon-hpc-lab · Dec 16, 2024 · 547c5a3 · 547c5a3
1 parent 502e99a
commit 547c5a3
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/inputs/exacb_turbulence.in b/inputs/exacb_turbulence.in
@@ -7,7 +7,7 @@ problem_id = turbulence
 
 <exacb/power>
 cycle_start = 50
-cycle_end   = 250
+cycle_stop  = 250
 
 <parthenon/time>
 cfl = 0.3               # The Courant, Friedrichs, & Lewy (CFL) Number

diff --git a/src/hydro/hydro.cpp b/src/hydro/hydro.cpp
@@ -681,14 +681,14 @@ std::shared_ptr<StateDescriptor> Initialize(ParameterInput *pin) {
   // EXACB custom modifications for power measurements
   if (pin->DoesBlockExist("exacb/power")) {
     const auto start = pin->GetOrAddInteger("exacb/power", "cycle_start", 50);
-    const auto end = pin->GetOrAddInteger("exacb/power", "cycle_stop", 250);
-    PARTHENON_REQUIRE_THROWS(end > start, "Measuring backwards does not make sense.");
+    const auto stop = pin->GetOrAddInteger("exacb/power", "cycle_stop", 250);
+    PARTHENON_REQUIRE_THROWS(stop > start, "Measuring backwards does not make sense.");
 
     const auto nlim = pin->GetInteger("parthenon/time", "nlim");
-    PARTHENON_REQUIRE_THROWS(nlim < 0 || nlim >= end, "end must be within nlim");
+    PARTHENON_REQUIRE_THROWS(nlim < 0 || nlim >= stop, "stop must be within nlim");
 
     pkg->AddParam<int>("exacb/power/start", start);
-    pkg->AddParam<int>("exacb/power/end", end);
+    pkg->AddParam<int>("exacb/power/stop", stop);
   }
 
   if (ProblemInitPackageData != nullptr) {

diff --git a/src/hydro/hydro_driver.cpp b/src/hydro/hydro_driver.cpp
@@ -13,6 +13,7 @@
 // Parthenon headers
 #include "amr_criteria/refinement_package.hpp"
 #include "bvals/comms/bvals_in_one.hpp"
+#include "globals.hpp"
 #include "prolong_restrict/prolong_restrict.hpp"
 #include <parthenon/parthenon.hpp>
 // AthenaPK headers
@@ -396,15 +397,16 @@ TaskCollection HydroDriver::MakeTaskCollection(BlockList_t &blocks, int stage) {
   const int num_partitions = pmesh->DefaultNumPartitions();
 
   // EXACB custom modifications for power measurements
-  if (stage == 1 && hydro_pkg->AllParams().hasKey("exacb/power/start")) {
+  if (stage == 1 && parthenon::Globals::my_rank == 0 &&
+      hydro_pkg->AllParams().hasKey("exacb/power/start")) {
     const auto start = hydro_pkg->Param<int>("exacb/power/start");
-    const auto end = hydro_pkg->Param<int>("exacb/power/end");
+    const auto stop = hydro_pkg->Param<int>("exacb/power/stop");
     std::ofstream outfile;
     if (start == tm.ncycle) {
       outfile.open("energy.times");
       outfile << "energy_start:";
     }
-    if (end == tm.ncycle) {
+    if (stop == tm.ncycle) {
       outfile.open("energy.times", std::ios_base::app);
       outfile << "energy_stop:";
     }