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

[AutoPGLE] Temporary disable pgle_test in the OSS.

Browse Source 
PiperOrigin-RevId: 697517161
This commit is contained in:
jax authors 2024-11-17 23:58:46 -08:00
parent 742cabc547
commit ed250b8983
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/pgle_test.py
View File

@ -43,6 +43,11 @@ jax.config.parse_flags_with_absl()
@jtu.pytest_mark_if_available('multiaccelerator')
# TODO(patrios): Remove this skip once b/379267258 is fixed.
@jtu.skip_under_pytest(
'This test requires specific XLA_FLAGS. However, pytest does not reload '
'modules between tests. So if another test is launched before this one '
'necessary XLA_FLAGS will not be re-used by the XLA.')
class PgleTest(jtu.JaxTestCase):
_dump_exit_stack: ExitStack | None = None