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

[JAX] Include the function name in the error message when hash/equality of a static argument fails.

Browse Source 
PiperOrigin-RevId: 395550728
This commit is contained in:
Peter Hawkins 2021-09-08 13:50:08 -07:00 committed by jax authors
parent b0e0e46109
commit 7f277068a9
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/api_test.py
View File

@ -439,10 +439,11 @@ class CPPJitTest(jtu.BufferDonationTestCase):
jitted_f(1, 1)
msg = ("Non-hashable static arguments are not supported. An error occured "
"while trying to hash an object of type <class 'numpy.ndarray'>, 1. "
"The error was:\nTypeError: unhashable type: 'numpy.ndarray'")
".*while trying to hash an object of type "
"<class 'numpy\\.ndarray'>, 1. The error was:\nTypeError: "
"unhashable type: 'numpy\\.ndarray'")
with self.assertRaisesRegex(ValueError, re.escape(msg)):
with self.assertRaisesRegex(ValueError, msg):
jitted_f(1, np.asarray(1))
class HashableWithoutEq: