fix bug gmres (#452)

Solves issue #451.
pyccel · Jan 22, 2025 · 74c1271 · 74c1271
1 parent 5ff3884
commit 74c1271
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/psydac/linalg/solvers.py b/psydac/linalg/solvers.py
@@ -1007,7 +1007,7 @@ def solve(self, b, out=None):
             pp += rp
 
             # new residual norm
-            res_sqr = r.dot(r)
+            res_sqr = r.dot(r).real
 
             niter += 1
 
@@ -1804,6 +1804,10 @@ def solve(self, b, out=None):
         r -= b
 
         am = sqrt(r.dot(r).real)
+        if am < tol:
+            self._info = {'niter': 1, 'success': am < tol, 'res_norm': am }
+            return x
+
         beta.append(am)
         r *= - 1 / am