mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-19 00:04:41 +08:00
fix type cast error in item() for bfloat16 (#339)
Co-authored-by: xuan <xuan@apple.com>
This commit is contained in:
@@ -102,6 +102,9 @@ class TestArray(mlx_tests.MLXTestCase):
|
||||
self.assertEqual(x.item(), 1)
|
||||
self.assertTrue(isinstance(x.item(), int))
|
||||
|
||||
x = mx.array(1, mx.bfloat16)
|
||||
self.assertEqual(x.item(), 1.0)
|
||||
|
||||
x = mx.array(1.0)
|
||||
self.assertEqual(x.size, 1)
|
||||
self.assertEqual(x.ndim, 0)
|
||||
|
Reference in New Issue
Block a user