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

Run compilation_cache_test.py using TFRT

Browse Source 
PiperOrigin-RevId: 426433041
This commit is contained in:
Skye Wanderman-Milne 2022-02-04 10:17:26 -08:00 committed by jax authors
parent f5107212db
commit 3fd91d6038
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/compilation_cache_test.py
View File

@ -43,8 +43,6 @@ class CompilationCacheTest(jtu.JaxTestCase):
super().setUp()
if jtu.device_under_test() != "tpu":
raise SkipTest("serialize executable only works on TPU")
if jax._src.lib.xla_bridge.get_backend().runtime_type == "tfrt":
raise SkipTest("the new TFRT runtime does not support serialization")
def tearDown(self):
super().tearDown()