New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[ONNX] Enable 'ExportOutput.save' for models larger than 2GB #107904
Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/107904
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (1 Unrelated Failure)As of commit 0f07ca7 with merge base 1ef4bd1 (): UNSTABLE - The following job failed but was likely due to flakiness present on trunk and has been marked as unstable:
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
ghstack-source-id: e92a457a3648d78f268649c6a256f31a70b87fa5 Pull Request resolved: #107904
Previously it fails during serialization, despite onnxscript graph_building managed to return ModelProto > 2GB. [ghstack-poisoned]
ghstack-source-id: addb87d18df4175ef7c29be205e30f346cee6c8a Pull Request resolved: #107904
… 2GB" Previously it fails during serialization, despite onnxscript graph_building managed to return ModelProto > 2GB. [ghstack-poisoned]
ghstack-source-id: 9593c65479fd4d7faa66925dd7028b072b38ef81 Pull Request resolved: #107904
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This looks good for now. We should revisit the serializer protocol again for 2.2.
Previously it fails during serialization, despite onnxscript graph_building managed to return ModelProto > 2GB. [ghstack-poisoned]
ghstack-source-id: b8c31b52901880eb231e124e19794d4f0a430cea Pull Request resolved: #107904
@pytorchbot merge |
Merge startedYour change will be merged once all checks pass (ETA 0-4 Hours). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Previously it fails during serialization, despite onnxscript graph_building managed to return ModelProto > 2GB. Pull Request resolved: #107904 Approved by: https://github.com/abock
Stack from ghstack (oldest at bottom):
Previously it fails during serialization, despite onnxscript graph_building managed to return ModelProto > 2GB.