Fix init from double (#2861)
Some checks failed
Build and Test / Check Lint (push) Has been cancelled
Build and Test / Linux (cpu, aarch64) (push) Has been cancelled
Build and Test / Linux (cpu, x86_64) (push) Has been cancelled
Build and Test / Linux (cuda-12.6, aarch64) (push) Has been cancelled
Build and Test / Linux (cuda-12.9, aarch64) (push) Has been cancelled
Build and Test / Linux (cuda-12.6, x86_64) (push) Has been cancelled
Build and Test / Linux (cuda-12.9, x86_64) (push) Has been cancelled
Build and Test / macOS (14.0) (push) Has been cancelled
Build and Test / macOS (15.0) (push) Has been cancelled
Build and Test / Build Documentation (push) Has been cancelled
Build and Test / Linux Fedora (aarch64) (push) Has been cancelled
Build and Test / Linux Fedora (x86_64) (push) Has been cancelled

This commit is contained in:
Awni Hannun
2025-12-03 06:08:11 -08:00
committed by GitHub
parent 193cdcd81a
commit cacbdbf995
2 changed files with 24 additions and 5 deletions

View File

@@ -434,6 +434,14 @@ class TestArray(mlx_tests.MLXTestCase):
x = mx.array([0, 4294967295], dtype=mx.float32)
self.assertTrue(np.array_equal(x, xnp))
def test_double_keeps_precision(self):
x = 39.14223403241
out = mx.array(x, dtype=mx.float64).item()
self.assertEqual(out, x)
out = mx.array([x], dtype=mx.float64).item()
self.assertEqual(out, x)
def test_construction_from_lists_of_mlx_arrays(self):
dtypes = [
mx.bool_,