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

jax_jit_test: inherit from JaxTestCase

Browse Source
This commit is contained in:
Jake VanderPlas 2022-01-24 15:32:59 -08:00
parent b8372b0ca2
commit 8cacad2e23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/jax_jit_test.py
View File

@ -39,7 +39,7 @@ def _cpp_device_put(value, device):
return jaxlib.jax_jit.device_put(value, config.x64_enabled, device)
class JaxJitTest(parameterized.TestCase):
class JaxJitTest(jtu.JaxTestCase):
def test_is_float_0(self):
self.assertTrue(