fix: set backward ad to True

DiffAPF · Apr 2, 2024 · e34d2dc · e34d2dc
1 parent 894a5b0
commit e34d2dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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))