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

Rollback fix repr() of jit-compiled functions

Browse Source 
PiperOrigin-RevId: 402621480
This commit is contained in:
jax authors 2021-10-12 12:04:21 -07:00
parent 7fa6b1b5fa
commit ef3e3b5593
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/api_test.py
View File

@ -80,14 +80,6 @@ class CPPJitTest(jtu.BufferDonationTestCase):
# Tensorflow.
return api._cpp_jit
@unittest.skipIf(jax._src.lib._xla_extension_version < 40,
"Test requires jaxlib 0.1.72")
def test_jit_func_repr(self):
def my_function():
return
jitted = jit(my_function)
self.assertEqual(repr(jitted), repr(my_function))
def test_jit_of_noncallable(self):
self.assertRaisesRegex(TypeError, "Expected a callable value.*",
lambda: self.jit(3))