OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[Export] Add math module for deserialization (#154643)

Browse Source 
Summary: As title

Test Plan: ci

Differential Revision: D75580646

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154643
Approved by: https://github.com/yushangdi
This commit is contained in:
Henry Hu 2025-05-30 17:29:22 +00:00 committed by PyTorch MergeBot
parent fc0135ca11
commit 802ffd06c8
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/_export/serde/serialize.py
View File

@ -1713,6 +1713,9 @@ class GraphModuleDeserializer(metaclass=Final):
elif serialized_target.startswith("torch"):
module = torch # type: ignore[misc]
serialized_target_names = serialized_target.split(".")[1:]
elif serialized_target.startswith("math"):
module = math # type: ignore[misc]
serialized_target_names = serialized_target.split(".")[1:]
elif serialized_target.startswith("#"):
return self.deserialize_extension_operator(serialized_target)
else: # TODO(zhxchen17) Don't catch all here.