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

Merge pull request #13948 from jakevdp:fix-finfo-test

Browse Source 
PiperOrigin-RevId: 501063882
This commit is contained in:
jax authors 2023-01-10 12:26:33 -08:00
commit 6d2c157354
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/dtypes_test.py
View File

@ -325,18 +325,21 @@ class DtypesTest(jtu.JaxTestCase):
self.assertEqual(info.nmant + info.nexp + 1, info.bits)
assertRepresentable(info.tiny)
assertRepresentable(info.smallest_subnormal)
assertRepresentable(info.max)
assertRepresentable(2.0 ** (info.maxexp - 1))
if dtype != np.float64: # avoid Python float overflows
assertInfinite(info.max * 2)
assertInfinite(2. ** info.maxexp)
assertZero(info.smallest_subnormal * 0.5)
# smallest_normal & smallest_subnormal added in numpy 1.22
if jtu.numpy_version() >= (1, 22, 0):
assertRepresentable(info.smallest_subnormal)
assertZero(info.smallest_subnormal * 0.5)
self.assertEqual(info.tiny, info.smallest_normal)
# Identities according to the documentation:
self.assertAllClose(info.resolution, make_val(10 ** -info.precision))
self.assertEqual(info.tiny, info.smallest_normal)
self.assertEqual(info.epsneg, make_val(2 ** info.negep))
self.assertEqual(info.eps, make_val(2 ** info.machep))
self.assertEqual(info.iexp, info.nexp)