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

fix conv2d bug + faster conv 1d (#2195)

Browse Source 
* fix conv2d bug + faster conv 1d

* revert sort + flaky test
This commit is contained in:
Awni Hannun
2025-05-18 06:05:11 -07:00
committed by GitHub
parent 0654543dcc
commit 8576e6fe36
6 changed files with 170 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/tests/test_vmap.py
View File

@@ -634,6 +634,7 @@ class TestVmap(mlx_tests.MLXTestCase):
self.assertEqual(fy.shape, (4, 5, 6, 7))
def test_leaks(self):
mx.synchronize()
if mx.metal.is_available():
mem_pre = mx.get_active_memory()
else: