diff --git a/test/test_autograd.py b/test/test_autograd.py index dbd1454ff7459..76fe65ea86b17 100644 --- a/test/test_autograd.py +++ b/test/test_autograd.py @@ -11755,7 +11755,7 @@ def forward(ctx, inp): @staticmethod def backward(ctx, grad): # Reentrant backward in child will take much longer. - reentrant_root.backward() + reentrant_root.backward(grad) return grad # Parent gpu graph.