From 7dcabe1f8d406b62df4141883e503cabb39b8d45 Mon Sep 17 00:00:00 2001
From: Anton Khirnov <anton@khirnov.net>
Date: Wed, 30 Jan 2019 12:51:18 +0100
Subject: mg2d: print stats properly for exact solves

---
 ell_grid_solve.h | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'ell_grid_solve.h')

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;
 
 /**
-- 
cgit v1.2.3