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

Further reduction tuning (#1349)

Browse Source 
* More reduction tuning
* Forgotten pdb
* Small column long row specialization
This commit is contained in:
Angelos Katharopoulos
2024-08-23 10:35:25 -07:00
committed by GitHub
parent da8deb2b62
commit b57a52813b
4 changed files with 77 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/tests/test_reduce.py
View File

@@ -43,10 +43,6 @@ class TestReduce(mlx_tests.MLXTestCase):
z_npy = np.sum(y_npy, axis=a) / 1000
z_mlx = mx.sum(y_mlx, axis=a) / 1000
mx.eval(z_mlx)
if not np.allclose(z_npy, np.array(z_mlx), atol=1e-4):
import pdb
pdb.set_trace()
self.assertTrue(
np.allclose(z_npy, np.array(z_mlx), atol=1e-4)
)