From bc5a072ebfe240cf8685b97bf86cf8ac1bb7797f Mon Sep 17 00:00:00 2001 From: RiyaP-QA Date: Sun, 28 Sep 2025 23:16:42 +0000 Subject: [PATCH] fixes import error 'functionalize' from functorch (#163746) Fixes #163637 Pull Request resolved: https://github.com/pytorch/pytorch/pull/163746 Approved by: https://github.com/malfet --- functorch/experimental/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/functorch/experimental/__init__.py b/functorch/experimental/__init__.py index 3941f6d96e1f..0500fc2c29d3 100644 --- a/functorch/experimental/__init__.py +++ b/functorch/experimental/__init__.py @@ -1,5 +1,5 @@ # PyTorch forward-mode is not mature yet -from functorch import functionalize from torch._functorch.apis import chunk_vmap from torch._functorch.batch_norm_replacement import replace_all_batch_norm_modules_ from torch._functorch.eager_transforms import hessian, jacfwd, jvp +from torch.func import functionalize