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

Disable lax_numpy test that seems to lead to nonterminating LLVM compilation.

Browse Source 
PiperOrigin-RevId: 466682802
This commit is contained in:
Peter Hawkins 2022-08-10 07:50:14 -07:00 committed by jax authors
parent 8b2e4f975c
commit 03590d86c0
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/lax_numpy_test.py
View File

@ -1603,6 +1603,8 @@ class LaxBackedNumpyTests(jtu.JaxTestCase):
for shape2 in all_shapes
for assume_unique in [False, True]
for return_indices in [False, True]))
# TODO(b/242043996): compilation doesn't terminate (LLVM bug?)
@jtu.skip_on_devices("cpu")
def testIntersect1d(self, shape1, dtype1, shape2, dtype2, assume_unique, return_indices):
rng = jtu.rand_default(self.rng())
args_maker = lambda: [rng(shape1, dtype1), rng(shape2, dtype2)]