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

Skip flaky test on TPU

Browse Source 
PiperOrigin-RevId: 499794466
This commit is contained in:
Adam Paszke 2023-01-05 03:09:22 -08:00 committed by jax authors
parent 008f35a6b4
commit f1635ca875
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/jaxpr_effects_test.py
View File

@ -597,6 +597,7 @@ class EffectOrderingTest(jtu.JaxTestCase):
jax.effects_barrier()
self.assertListEqual(log, [2., 3.])
@jtu.skip_on_devices("tpu")
def test_ordered_effect_remains_ordered_across_multiple_devices(self):
# TODO(sharadmv): enable this test on GPU and TPU when backends are
# supported