frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Dynamo] Fix inspect.getattr_static doesn't work well for torch.utils._cxx_pytree.PyTreeSpec (#120812)

Browse Source 
Fixes #118793

Pull Request resolved: https://github.com/pytorch/pytorch/pull/120812
Approved by: https://github.com/zou3519
This commit is contained in:
Yanbo Liang
2024-03-05 09:05:20 +00:00
committed by PyTorch MergeBot
parent 311cc564f6
commit 46c9d646dd
12 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_namedtuple
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec0
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec1
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec2
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec3
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec4
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec5
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec6
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec7
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec8
View File

0
test/dynamo_expected_failures/TestCxxPytree.test_pytree_serialize_spec9
View File

7
torch/_dynamo/variables/user_defined.py
View File

@ -19,6 +19,11 @@ try:
except ModuleNotFoundError:
np = None
try:
from torch.utils._cxx_pytree import PyTreeSpec
except ImportError:
PyTreeSpec = type(None)
import torch._dynamo.config
import torch.nn
@ -697,7 +702,7 @@ class UserDefinedObjectVariable(UserDefinedVariable):
def _getattr_static(self, name):
if (
isinstance(self.value, torch.nn.Module)
isinstance(self.value, (torch.nn.Module, PyTreeSpec))
or "__slots__" in self.value.__class__.__dict__
or type(self.value) == threading.local
):