Add adaptivity_cpu_time output to local adaptivity metrics too

precice · Feb 11, 2025 · de8c5a4 · de8c5a4
1 parent 9fcfca9
commit de8c5a4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/micro_manager/adaptivity/global_adaptivity.py b/micro_manager/adaptivity/global_adaptivity.py
@@ -179,7 +179,7 @@ def get_full_field_micro_output(self, micro_output: list) -> list:
 
         return micro_sims_output
 
-    def log_metrics(self, n: int, adaptivity_cpu_time) -> None:
+    def log_metrics(self, n: int, adaptivity_cpu_time: float) -> None:
         """
         Log metrics for global adaptivity.
 

diff --git a/micro_manager/adaptivity/local_adaptivity.py b/micro_manager/adaptivity/local_adaptivity.py
@@ -138,14 +138,16 @@ def get_full_field_micro_output(self, micro_output: list) -> list:
 
         return micro_sims_output
 
-    def log_metrics(self, n: int) -> None:
+    def log_metrics(self, n: int, adaptivity_cpu_time: float) -> None:
         """
         Log metrics for local adaptivity.
 
         Parameters
         ----------
         n : int
             Current time step
+        adaptivity_cpu_time : float
+            CPU time taken for adaptivity calculation
         """
         # MPI Gather is necessary as local adaptivity only stores local data
         local_active_sims = np.count_nonzero(self._is_sim_active)
@@ -155,12 +157,13 @@ def log_metrics(self, n: int) -> None:
         global_inactive_sims = self._comm.gather(local_inactive_sims)
 
         self._metrics_logger.log_info_one_rank(
-            "{},{},{},{},{}".format(
+            "{},{},{},{},{},{}".format(
                 n,
                 np.mean(global_active_sims),
                 np.mean(global_inactive_sims),
                 np.max(global_active_sims),
                 np.max(global_inactive_sims),
+                adaptivity_cpu_time,
             )
         )