diff --git a/tests/test_autograd.py b/tests/test_autograd.py index 5171c4f..28d9259 100644 --- a/tests/test_autograd.py +++ b/tests/test_autograd.py @@ -427,4 +427,4 @@ def test_matmullt( ) if req_grad[2]: - torch.testing.assert_allclose(gradBias1, gradBias2) + torch.testing.assert_allclose(gradBias1, gradBias2, atol=0.18, rtol=0.3)