From 4ff85801e3f199078808a221b011d5844e294ab0 Mon Sep 17 00:00:00 2001 From: Yan Wang Date: Fri, 18 Oct 2024 13:36:38 +0200 Subject: [PATCH] fix rebase functional-autograd-checkpoint --- thunder/torch/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/thunder/torch/__init__.py b/thunder/torch/__init__.py index ed69b4f096..9406e87a84 100644 --- a/thunder/torch/__init__.py +++ b/thunder/torch/__init__.py @@ -5262,8 +5262,8 @@ def _backward_checkpoint( ) -> tuple[None | TensorLike, ...]: from thunder.core.transforms import vjp - result = vjp(function)(args, grad_outputs, **kwargs) - return result + result, grads = vjp(function)(args, grad_outputs, **kwargs) + return grads #