Skip to content

Conversation

aporialiao
Copy link
Contributor

Summary:

  • Using refactored ModelInput generation from D71556886 for dynamic sharding unit test. This will be important for CW sharding since existing input generation call here is insufficient for handling multiple shards per table.

  • Add additional docs explaining uses for some Dynamic Sharding API

  • Rename ModelInput kjt creation method to indicate static use outside of ModelInput class

Reviewed By: TroyGarden

Differential Revision: D72798357

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

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

aporialiao added a commit to aporialiao/torchrec that referenced this pull request Apr 10, 2025
Summary:

* Using refactored `ModelInput` generation from D71556886 for dynamic sharding unit test. This will be important for CW sharding since existing input generation call here is insufficient for handling multiple shards per table. 

* Add additional docs explaining uses for some Dynamic Sharding API

* Rename ModelInput kjt creation method to indicate static use outside of ModelInput class

Reviewed By: TroyGarden

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

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

aporialiao added a commit to aporialiao/torchrec that referenced this pull request Apr 10, 2025
Summary:

* Using refactored `ModelInput` generation from D71556886 for dynamic sharding unit test. This will be important for CW sharding since existing input generation call here is insufficient for handling multiple shards per table. 

* Add additional docs explaining uses for some Dynamic Sharding API

* Rename ModelInput kjt creation method to indicate static use outside of ModelInput class

Reviewed By: TroyGarden

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

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

Summary:
Pull Request resolved: meta-pytorch#2875

* Using refactored `ModelInput` generation from D71556886 for dynamic sharding unit test. This will be important for CW sharding since existing input generation call here is insufficient for handling multiple shards per table.

* Add additional docs explaining uses for some Dynamic Sharding API

* Rename ModelInput kjt creation method to indicate static use outside of ModelInput class

Reviewed By: TroyGarden

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

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

aporialiao added a commit to aporialiao/torchrec that referenced this pull request Apr 10, 2025
Summary:

* Using refactored `ModelInput` generation from D71556886 for dynamic sharding unit test. This will be important for CW sharding since existing input generation call here is insufficient for handling multiple shards per table. 

* Add additional docs explaining uses for some Dynamic Sharding API

* Rename ModelInput kjt creation method to indicate static use outside of ModelInput class

Reviewed By: TroyGarden

Differential Revision: D72798357
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