Register lsqr code 2 as not converged

tum-pbs · Jan 10, 2025 · ee24fcf · ee24fcf
1 parent cd4e909
commit ee24fcf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phiml/backend/_linalg.py b/phiml/backend/_linalg.py
@@ -406,7 +406,7 @@ def count_callback(x_n):  # called after each step, not with x0
             else:
                 messages.append(f"Least squares problem not solved (code {ret_val})")
             xs.append(x)
-            converged.append(ret_val in {0, 1, 2})
+            converged.append(ret_val in {0, 1})
             diverged.append(ret_val not in {0, 1, 2})
             iterations[bi] = n_iter
             residual.append(lin_b @ x - y[bi])