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

add test for device constant compile options bug

Browse Source
This commit is contained in:
Matthew Johnson 2019-02-25 13:48:01 -08:00
parent 8c5e41042b
commit a37703552f
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/api_test.py
View File

@ -337,6 +337,9 @@ class APITest(jtu.JaxTestCase):
f(2)
assert len(effects) == 3
def test_large_device_constant(self):
jit(lambda x: x)(np.zeros(int(2e6))) # doesn't crash
if __name__ == '__main__':
absltest.main()