diff --git a/test/test_tensor.py b/test/test_tensor.py index fb9bbc72cb..2ccf617aa4 100644 --- a/test/test_tensor.py +++ b/test/test_tensor.py @@ -152,6 +152,27 @@ class TestTinygrad(unittest.TestCase): for x,y in zip(test_tinygrad(), test_pytorch()): np.testing.assert_allclose(x, y, atol=1e-5, rtol=1e-6) + @unittest.expectedFailure + def test_const_backward_pass(self): + init = 3.5 + + def test_pytorch(): + w1 = torch.tensor(init, requires_grad=True) + w2 = torch.tensor(init, requires_grad=True) + out = w1.add(w2) + out.backward() + return w1.grad, w2.grad + + def test_tinygrad(): + w1 = Tensor(init, requires_grad=True) + w2 = Tensor(init, requires_grad=True) + out = w1.add(w2) + out.backward() + return w1.grad.numpy(), w2.grad.numpy() + + for x, y in zip(test_tinygrad(), test_pytorch()): + np.testing.assert_allclose(x, y, atol=1e-5) + def test_nograd(self): x = Tensor(x_init, requires_grad=False) m = Tensor(m_init, requires_grad=False)