zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make the max nanpropagation test more meaningful for integer types

Browse Source
This commit is contained in:
Joona Havukainen
2025-07-08 16:42:19 -07:00
parent 8b15773206
commit aca7fac9ef
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_reduce.py
View File

@@ -167,7 +167,7 @@ class TestReduce(mlx_tests.MLXTestCase):
for dtype in dtypes: for dtype in dtypes:
with self.subTest(dtype=dtype): with self.subTest(dtype=dtype):
x = (mx.random.normal((4, 4))).astype(getattr(mx, dtype)) x = (mx.random.normal((4, 4)) * 10).astype(getattr(mx, dtype))
indices = mx.random.randint(0, 4, shape=(6,)).reshape(3, 2) indices = mx.random.randint(0, 4, shape=(6,)).reshape(3, 2)
for idx in indices: for idx in indices:
x[*idx] = mx.nan x[*idx] = mx.nan