geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

test_gradient_through_clone (#16203)

Browse Source 
backward through clone crashes now
This commit is contained in:
chenyu
2026-05-14 19:26:47 -04:00
committed by GitHub
parent d4dcd8487b
commit 09096ea565
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/unit/test_gradient.py
+15
View File
@@ -69,6 +69,21 @@ class TestTensorGradient(unittest.TestCase):
np.testing.assert_allclose(x.grad.numpy(), [2.0+3.0+2*3.0])
self.assertIs(x.grad, old_grad)
def test_gradient_through_clone(self):
src = Tensor([1.0, 2.0, 3.0, 4.0])
x = src.clone().requires_grad_(True)
(x * 2.0).sum().backward()
np.testing.assert_allclose(x.grad.numpy(), [2.0, 2.0, 2.0, 2.0])
self.assertIsNone(src.grad)
src = Tensor([1.0, 2.0, 3.0, 4.0], requires_grad=True)
x = src.clone().requires_grad_(True)
try:
(x * 2.0).sum().backward()
except RuntimeError:
# TODO: this crashes now
pass
def test_gradient_through_chained_unrealized_setitem(self):
g1 = Tensor.zeros(4).contiguous()
g1[2] = Tensor(1.0)