mg2d: print stats properly for exact solves

1 files changed, 9 insertions, 1 deletions

diff --git a/ell_grid_solve.h b/ell_grid_solve.h
index 4ee6cb4..5455c13 100644
--- a/ell_grid_solve.h
+++ b/ell_grid_solve.h
@@ -55,7 +55,7 @@ enum EGSType {
     /**
      * Solve the equation exactly by contructing a linear system and solving it with LAPACK.
      *
-     * solver_data is NULL
+     * solver_data is EGSExactContext
      * mg2di_egs_solve() solves the discretized system exactly (up to roundoff error)
      */
     EGS_SOLVER_EXACT,
@@ -87,6 +87,13 @@ typedef struct EGSRelaxContext {
     int64_t time_correct;
 } EGSRelaxContext;
 
+typedef struct EGSExactContext {
+    int64_t count_mat_construct;
+    int64_t time_mat_construct;
+    int64_t count_lin_solve;
+    int64_t time_lin_solve;
+} EGSExactContext;
+
 typedef struct EGSContext {
     enum EGSType solver_type;
 
@@ -202,6 +209,7 @@ typedef struct EGSContext {
     int64_t count_boundaries;
     int64_t time_res_calc;
     int64_t count_res;
+    int64_t time_total;
 } EGSContext;
 
 /**