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

[host_callback] Skip test that only works in legacy mode.

Browse Source 
The jax.experimental.host_callback module is deprecated and will be removed.

See https://github.com/google/jax/issues/20385.

PiperOrigin-RevId: 680988939
This commit is contained in:
George Necula 2024-10-01 06:12:55 -07:00 committed by jax authors
parent 98b72b17f9
commit a644e23a4b
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/host_callback_test.py
View File

@ -585,6 +585,7 @@ class HostCallbackTapTest(jtu.JaxTestCase):
# See also b/160692602.
raise SkipTest("concurrent id_tap not supported on CPU, GPU")
self.supported_only_in_legacy_mode()
received = set()
count = 5