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

Mark error_check_test as thread-unsafe.

Browse Source 
This is a temporary fix to make CI green, but the right thing to do is to fix the implementation to be thread-safe.

PiperOrigin-RevId: 730445067
This commit is contained in:
Peter Hawkins 2025-02-24 07:35:47 -08:00 committed by jax authors
parent c17ea805f3
commit aeff675eda
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/error_check_test.py
View File

@ -28,6 +28,7 @@ JaxValueError = error_check.JaxValueError
config.parse_flags_with_absl()
@jtu.thread_unsafe_test_class() # TODO(b/398772684): fix thread safety
@jtu.with_config(jax_check_tracer_leaks=True)
class ErrorCheckTests(jtu.JaxTestCase):