Commit 9c44b34
committed
BUG: limit the range of elements in test_{eig,eigvals}
Otherwise torch fails to converge if the matrix is too ill-conditioned.
Notice 6.7e+111, 6.9e+128, 7e+90 values in this failing example:
```
> res = linalg.eigvals(x)
E torch._C._LinAlgError: torch.linalg.eigvals: (Batch element 1): The algorithm failed to converge because the input matrix is ill-conditioned or has too many repeated eigenvalues (error code: 6).
E Falsifying example: test_eigvals(
E x=tensor([[[ 0.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00],
E [ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00],
E [ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00],
E [ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00],
E [ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00],
E [ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00]],
E
E [[ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00],
E [6.7525e+111, 2.0000e+00, 2.0000e+00, 6.9317e+128, 2.0000e+00,
E 2.0000e+00],
E [ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00],
E [ 2.0000e+00, 0.0000e+00, 2.0000e+00, 2.0000e+00, 7.5592e+90,
E 2.0000e+00],
E [ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00],
E [ 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00, 2.0000e+00,
E 2.0000e+00]]], dtype=torch.float64),
E )
array_api_tests/test_linalg.py:375: _LinAlgError
```1 parent 41379d1 commit 9c44b34
2 files changed
+10
-7
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
331 | 331 | | |
332 | 332 | | |
333 | 333 | | |
334 | | - | |
335 | | - | |
336 | | - | |
337 | | - | |
338 | | - | |
| 334 | + | |
| 335 | + | |
| 336 | + | |
| 337 | + | |
| 338 | + | |
| 339 | + | |
| 340 | + | |
| 341 | + | |
339 | 342 | | |
340 | 343 | | |
341 | 344 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
335 | 335 | | |
336 | 336 | | |
337 | 337 | | |
338 | | - | |
| 338 | + | |
339 | 339 | | |
340 | 340 | | |
341 | 341 | | |
| |||
370 | 370 | | |
371 | 371 | | |
372 | 372 | | |
373 | | - | |
| 373 | + | |
374 | 374 | | |
375 | 375 | | |
376 | 376 | | |
| |||
0 commit comments