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

Remove Hazard tracking with Fences (#1509)

Browse Source 
* remove hazard tracking

* with fence map

* no hazard tracking with fences

* nits

* fix fence retain

* cleanup

* fix quantized rebase
This commit is contained in:
Awni Hannun
2024-10-21 19:33:32 -07:00
committed by GitHub
parent d15fa13daf
commit c26208f67d
25 changed files with 268 additions and 299 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/tests/test_compile.py
View File

@@ -769,7 +769,6 @@ class TestCompile(mlx_tests.MLXTestCase):
out = mx.compile(fn)(a, b)
expected = fn(a, b)
print((out - expected).abs().max())
self.assertTrue(mx.allclose(out, expected))
def test_compile_many_inputs(self):