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

Disable breaking testAutoPgleWithPersistentCache. The test is unstable and it seems to be caused by temporary directory not being properly allocated/released in case of parallel execution of tests via pytest-xdist.

Browse Source 
Way to reproduce:
```
python3.10 -m pytest -n 2 --tb=short --maxfail=20 tests/pgle_test.py
```

Note, if `-n` argument is 1 (i.e. behave like regular single-threaded pytest) the tests pass. It also passes in bazel, since bazel better isolates different tests from each other.

PiperOrigin-RevId: 692707058
This commit is contained in:
Vadym Matsishevskyi 2024-11-03 08:38:52 -08:00 committed by jax authors
parent ec39b592f7
commit d4ddabb1d1
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/pgle_test.py
View File

@ -177,6 +177,7 @@ class PgleTest(jtu.JaxTestCase):
self.assertArraysEqual(compiled(x), expected)
self.assertEqual(cache_miss_count[0], 0)
@unittest.skip("Test failing in CI")
def testAutoPgleWithPersistentCache(self):
its = 50
mesh = jtu.create_mesh((2,), ('x',))