[Easy][pytree] enable CXX pytree under torch::deploy (#130144)

Pull Request resolved: #130144 Approved by: https://github.com/zou3519 ghstack dependencies: #130895, #130139
pytorch · Jul 21, 2024 · 1439bd3 · 1439bd3
1 parent ddde9dd
commit 1439bd3
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/torch/utils/_cxx_pytree.py b/torch/utils/_cxx_pytree.py
@@ -29,11 +29,6 @@
 )
 from typing_extensions import deprecated
 
-import torch
-
-if torch._running_with_deploy():  # type: ignore[no-untyped-call]
-    raise ImportError("C++ pytree utilities do not work with torch::deploy.")
-
 import optree
 from optree import PyTreeSpec  # direct import for type annotations