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

update test of dlpack error message

Browse Source
This commit is contained in:
Matthew Johnson 2021-10-07 19:14:13 -07:00
parent efa5edfd39
commit ef710ec1f6
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/array_interoperability_test.py
View File

@ -167,7 +167,7 @@ class DLPackTest(jtu.JaxTestCase):
backend = xla_bridge.get_backend()
client = getattr(backend, "client", backend)
regex_str = (r'Unimplemented: Only DLPack tensors with trivial \(compact\) '
regex_str = (r'UNIMPLEMENTED: Only DLPack tensors with trivial \(compact\) '
r'striding are supported')
with self.assertRaisesRegex(RuntimeError, regex_str):
xla_client._xla.dlpack_managed_tensor_to_buffer(