logging fixes (#901)

illinois-ceesd · May 9, 2023 · 07c324c · 07c324c
1 parent bd2a79a
commit 07c324c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/examples/combozzle-mpi.py b/examples/combozzle-mpi.py
@@ -1234,11 +1234,6 @@ def dummy_rhs(t, state):
     my_write_restart(step=current_step, t=current_t, state=final_fluid_state,
                      temperature_seed=tseed)
 
-    if logmgr:
-        logmgr.close()
-    elif use_profiling:
-        print(actx.tabulate_profiling_data())
-
     comm.Barrier()
 
     finish_tol = 1e-16
@@ -1256,6 +1251,11 @@ def dummy_rhs(t, state):
 
     comm.Barrier()
 
+    if logmgr:
+        logmgr.close()
+    elif use_profiling:
+        print(actx.tabulate_profiling_data())
+
 
 if __name__ == "__main__":
     import argparse

diff --git a/examples/heat-source-mpi.py b/examples/heat-source-mpi.py
@@ -179,14 +179,14 @@ def rhs(t, u):
             set_dt(logmgr, dt)
             logmgr.tick_after()
 
-    if logmgr:
-        logmgr.close()
-
     final_answer = actx.to_numpy(op.norm(dcoll, u, np.inf))
     resid = abs(final_answer - 0.0002062062188374177)
     if resid > 1e-15:
         raise ValueError(f"Run did not produce the expected result {resid=}")
 
+    if logmgr:
+        logmgr.close()
+
 
 if __name__ == "__main__":
     import argparse