From 0baa72a2d3b0a3a7303f42f9ce84d145d126b7c5 Mon Sep 17 00:00:00 2001
From: Anton Khirnov <anton@khirnov.net>
Date: Sun, 22 Apr 2018 17:43:02 +0200
Subject: nlsolve: account for early abort in counting total solve time

---
 nlsolve.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nlsolve.c b/nlsolve.c
index 2137767..9366605 100644
--- a/nlsolve.c
+++ b/nlsolve.c
@@ -212,7 +212,8 @@ int tdi_nlsolve_solve(NLSolveContext *ctx, NLEqCallback eq_eval,
         } else if ((fast_abort && fabs(s->delta[max_idx]) > 1e6) ||
                    s->delta[max_idx] > 1e18) {
             tdi_log(&ctx->logger, 2, "max(delta) %g, aborting\n", s->delta[max_idx]);
-            return -EDOM;
+            ret = -EDOM;
+            goto finish;
         }
 
         cblas_daxpy(s->solve_order, 1.0, s->delta, 1, coeffs, 1);
-- 
cgit v1.2.3