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

Add version guard to compilation cache test.

Browse Source 
PiperOrigin-RevId: 525572568
This commit is contained in:
Peter Hawkins 2023-04-19 15:49:55 -07:00 committed by jax authors
parent fb5664d580
commit 34fd4a1562
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/compilation_cache_test.py
View File

@ -197,6 +197,8 @@ class CompilationCacheTest(jtu.JaxTestCase):
cc.get_cache_key(computation2, compile_options, backend),
)
@unittest.skipIf(jax._src.lib.version < (0, 4, 9),
"Test requires jaxlib 0.4.9")
@parameterized.parameters([False, True])
def test_identical_computations_different_metadata(self, include_metadata):
f = lambda x, y: lax.mul(lax.add(x, y), 2)