zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-19 08:29:35 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

xor op on arrays (#1875)

Browse Source
This commit is contained in:
Alex Barron
2025-02-17 00:24:53 -08:00
committed by GitHub
parent 5274c3c43f
commit 4c1dfa58b7
2 changed files with 38 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/tests/test_array.py
View File

@@ -1725,6 +1725,7 @@ class TestArray(mlx_tests.MLXTestCase):
self.assertEqual((mx.array(True) | False).item(), True)
self.assertEqual((mx.array(False) | False).item(), False)
self.assertEqual((~mx.array(False)).item(), True)
self.assertEqual((mx.array(False) ^ True).item(), True)
def test_inplace(self):
iops = [
@@ -1734,6 +1735,7 @@ class TestArray(mlx_tests.MLXTestCase):
"__ifloordiv__",
"__imod__",
"__ipow__",
"__ixor__",
]
for op in iops:
@@ -1773,6 +1775,10 @@ class TestArray(mlx_tests.MLXTestCase):
b @= a
self.assertTrue(mx.array_equal(a, b))
a = mx.array(False)
a ^= True
self.assertEqual(a.item(), True)
def test_inplace_preserves_ids(self):
a = mx.array([1.0])
orig_id = id(a)