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

Primitive's VJP takes outputs as input (#475)

Browse Source 
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
This commit is contained in:
Awni Hannun
2024-01-16 19:03:53 -08:00
committed by GitHub
parent d8fabaa12b
commit a2bf7693dd
5 changed files with 205 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/mlx_tests.py
View File

@@ -71,3 +71,5 @@ class MLXTestCase(unittest.TestCase):
elif not isinstance(expected, mx.array):
expected = mx.array(expected)
self.assertTrue(mx.allclose(mx_res, expected, rtol=rtol, atol=atol))
else:
self.assertTrue(mx.allclose(mx_res, expected, rtol=rtol, atol=atol))