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

Fix imag() vjp (#2367)

Browse Source
This commit is contained in:
Angelos Katharopoulos
2025-07-14 13:11:16 -07:00
committed by GitHub
parent 2d3c26c565
commit 5201df5030
2 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_autograd.py
View File

@@ -606,7 +606,7 @@ class TestAutograd(mlx_tests.MLXTestCase):
x = mx.array([0.0 + 1j, 1.0 + 0.0j, 0.5 + 0.5j])
dfdx = mx.grad(fun)(x)
self.assertTrue(mx.allclose(dfdx, -2j * mx.ones_like(x)))
self.assertTrue(mx.allclose(dfdx, 2j * mx.ones_like(x)))
def test_flatten_unflatten_vjps(self):
def fun(x):