Skip to content

Introduce Serializer in TorchRec for torch.export PEA #1848

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

Closed
wants to merge 1 commit into from

Conversation

PaulZhang12
Copy link
Contributor

Summary: Adding a serializer class to support thrift serialization of PEA configs for regenerating the eager module after torch.export through saving the metadata as a bytes buffer in the module.

Differential Revision: D55661704

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Apr 4, 2024
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D55661704

Summary:

Adding a serializer class to support thrift serialization of PEA configs for regenerating the eager module after torch.export through saving the metadata as a bytes buffer in the module.

Differential Revision: D55661704
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D55661704

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants