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

fix power (#2523)

Browse Source
This commit is contained in:
Awni Hannun
2025-08-21 06:46:01 -07:00
committed by GitHub
parent 0c5fc63a36
commit e843c4d8d5
4 changed files with 24 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/tests/test_ops.py
View File

@@ -3068,6 +3068,13 @@ class TestOps(mlx_tests.MLXTestCase):
d = mx.where(c, a[1:], b)
self.assertTrue(mx.all(d == 1.0))
def test_integer_power(self):
x = mx.power(2, mx.array([8, 8, 8, 8, 8, 8, 8, 8]))
self.assertTrue(mx.all(x == 256))
# Doesn't hang
x = mx.power(2, -1)
class TestBroadcast(mlx_tests.MLXTestCase):
def test_broadcast_shapes(self):