Print stats on terminate

2 files changed, 13 insertions, 0 deletions

diff --git a/schedule.ccl b/schedule.ccl
index 88a30cb..673a151 100644
--- a/schedule.ccl
+++ b/schedule.ccl
@@ -54,6 +54,10 @@ if (CCTK_Equals(lapse_evolution_method, "maximal_axi_mg")) {
         LANG: C
     } ""
 
+    SCHEDULE msa_mg_terminate_print_stats IN CCTK_TERMINATE {
+        LANG: C
+    } ""
+
     STORAGE: lapse_mg[2]
     STORAGE: lapse_mg_eval
     STORAGE: lapse_prev0
diff --git a/src/maximal_slicing_axi_mg.c b/src/maximal_slicing_axi_mg.c
index 833fbbf..ab22685 100644
--- a/src/maximal_slicing_axi_mg.c
+++ b/src/maximal_slicing_axi_mg.c
@@ -1016,3 +1016,12 @@ void maximal_slicing_axi_mg_modify_diss(CCTK_ARGUMENTS)
             epsdis[idx_dst] = 0.0;
         }
 }
+
+void msa_mg_terminate_print_stats(CCTK_ARGUMENTS)
+{
+    const int reflevel = ctz(cctkGH->cctk_levfac[0]);
+    if (reflevel == 0 && ms) {
+        print_stats(ms);
+        context_free(&ms);
+    }
+}