From e34d2dcaffc949643644429fa8e366ab94646a39 Mon Sep 17 00:00:00 2001 From: Chin-Yun Yu Date: Tue, 2 Apr 2024 11:08:39 +0100 Subject: [PATCH] fix: set backward ad to True --- tests/test_grad.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_grad.py b/tests/test_grad.py index 2631134..d3775f7 100644 --- a/tests/test_grad.py +++ b/tests/test_grad.py @@ -63,5 +63,5 @@ def test_low_order_cpu( at.requires_grad = at_requires_grad rt.requires_grad = rt_requires_grad - assert gradcheck(compressor_core, (x, zi, at, rt), check_forward_ad=True, check_backward_ad=False) + assert gradcheck(compressor_core, (x, zi, at, rt), check_forward_ad=True) assert gradgradcheck(compressor_core, (x, zi, at, rt))