Move get_unwrapped outside of disable_functorch (#121849)

Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: #121849
Approved by: https://github.com/albanD

Author: ezyang

torch/_subclasses/meta_utils.py

@@ -632,8 +632,9 @@ def _to_fake_tensor(t):
                             r = _add_batch_dim(ft, bdim, lvl)
                         elif is_gradtrackingtensor(t):
                             disable_functorch = torch._C._DisableFuncTorch
+                            ft = get_unwrapped(t)
                             with disable_functorch():
-                                ft = _to_fake_tensor(get_unwrapped(t))
+                                ft = _to_fake_tensor(ft)
                             lvl = torch._C._functorch.maybe_get_level(t)
                             r = torch._C._functorch._wrap_for_grad(ft, lvl)