mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Disable lobpcg F32 consistency test on GPU.

Browse Source 
The test appears to be failing at least some of the time

Typical failure:

=================================== FAILURES ===================================
___________ F32LobpcgTest.testLobpcgConsistencyF32cluster_k_2__n100 ____________
[gw12] linux -- Python 3.9.12 /usr/local/bin/python3.9
tests/lobpcg_test.py:370: in testLobpcgConsistencyF32
    self.checkLobpcgConsistency(matrix_name, n, k, m, jnp.float32)
tests/lobpcg_test.py:203: in checkLobpcgConsistency
    self.assertLess(
E   AssertionError: DeviceArray(20, dtype=int32, weak_type=True) not less than 20 : expected early convergence iters 20 < max 20
This commit is contained in:
Peter Hawkins 2022-07-01 09:47:49 -04:00
parent 7b59bd02ae
commit 1d83f7dd6f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/lobpcg_test.py
View File

@ -366,6 +366,7 @@ class F32LobpcgTest(LobpcgTest):
linalg.lobpcg_standard(A[:50, :50], X[:50])
@parameterized.named_parameters(_make_concrete_cases(f64=False))
@jtu.skip_on_devices("gpu")
def testLobpcgConsistencyF32(self, matrix_name, n, k, m):
self.checkLobpcgConsistency(matrix_name, n, k, m, jnp.float32)