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

only test if omnistaging is enabled

Browse Source
This commit is contained in:
Matthew Johnson 2020-09-21 19:33:14 -07:00
parent 6b60267e55
commit 34e0460961
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/api_test.py
View File

@ -1798,6 +1798,9 @@ class APITest(jtu.JaxTestCase):
f() # doesn't crash
def test_xla_computation_zeros_doesnt_device_put(self):
if not config.omnistaging_enabled:
raise unittest.SkipTest("test is omnistaging-specific")
count = 0
def device_put_and_count(*args, **kwargs):
nonlocal count