make check for infinite eigenvalues more reliable

diprism · Sep 12, 2024 · d526c3a · d526c3a
1 parent 7da3552
commit d526c3a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/fggs/semirings.py b/fggs/semirings.py
@@ -156,9 +156,10 @@ def solve_thunks(self,
         b = make_b()
         try:
             a.neg_().diagonal().add_(1)
-            x = torch.linalg.solve(a, b)
-            if torch.all(torch.copysign(torch.tensor(1.), x) > 0.): # catches -0.0
-                return x
+            if not torch.any(torch.isinf(a)):
+                x = torch.linalg.solve(a, b)
+                if torch.all(x >= 0.):
+                    return x
         except RuntimeError as e:
             if '(Cannot allocate memory)' in str(e): raise
             pass