Skip to content

Conversation

PaulZhang12
Copy link
Contributor

Summary: Fixing device issues with sharding TorchRec inference modules on meta device

Reviewed By: IvanKobzarev

Differential Revision: D56020810

@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 15, 2024
@facebook-github-bot
Copy link
Contributor

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

PaulZhang12 added a commit to PaulZhang12/torchrec that referenced this pull request Apr 15, 2024
Summary:

Fixing device issues with sharding TorchRec inference modules on meta device

Reviewed By: IvanKobzarev

Differential Revision: D56020810
Summary:

Fixing device issues with sharding TorchRec inference modules on meta device

Reviewed By: IvanKobzarev

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

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

@facebook-github-bot
Copy link
Contributor

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

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