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

[export] Fix public bindings (#159109)

Browse Source 
Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159109
Approved by: https://github.com/jbschlosser
This commit is contained in:
angelayi
2025-07-25 18:18:47 +00:00
committed by PyTorch MergeBot
parent 4c0d5ad4be
commit 8a0508335f
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/allowlist_for_publicAPI.json
View File

@ -2650,11 +2650,6 @@
"torch.export.graph_signature": [
"TokenArgument"
],
"torch.export.pt2_archive": [
"PT2ArchiveWriter",
"PT2ArchiveReader",
"is_pt2_package"
],
"torch.fx.experimental.shape_inference.infer_shape": [
"DimDynamic",
"FakeTensorMode",

5
torch/export/pt2_archive/_package.py
View File

@ -216,6 +216,11 @@ class PT2ArchiveReader:
return self.archive_file.get_all_records()
is_pt2_package.__module__ = "torch.export.pt2_archive"
PT2ArchiveWriter.__module__ = "torch.export.pt2_archive"
PT2ArchiveReader.__module__ = "torch.export.pt2_archive"
def _package_aoti_files(
archive_writer: PT2ArchiveWriter,
aoti_files: Optional[AOTI_FILES],