Skip to content

Support ebc for kv zch #3106

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

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Conversation

EddyLXJ
Copy link

@EddyLXJ EddyLXJ commented Jun 17, 2025

Differential Revision: D76636927

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

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

@facebook-github-bot
Copy link
Contributor

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

EddyLXJ added a commit to EddyLXJ/torchrec that referenced this pull request Jun 17, 2025
Summary:
Pull Request resolved: pytorch#3106

# Change logs
1. add ZeroCollisionKeyValueEmbeddingBag emb lookup
2. address existing unit test missing for ebc ssd offloading
3. add new ut for kv zch embedding bag module

Embedding bag update is same as embedding update. Details can be found in D73567631

Differential Revision: D76636927
@EddyLXJ EddyLXJ force-pushed the export-D76636927 branch from 8447505 to b3622b7 Compare June 17, 2025 23:51
@facebook-github-bot
Copy link
Contributor

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

EddyLXJ added a commit to EddyLXJ/torchrec that referenced this pull request Jun 18, 2025
Summary:
Pull Request resolved: pytorch#3106

# Change logs
1. add ZeroCollisionKeyValueEmbeddingBag emb lookup
2. address existing unit test missing for ebc ssd offloading
3. add new ut for kv zch embedding bag module

Embedding bag update is same as embedding update. Details can be found in D73567631

Reviewed By: duduyi2013

Differential Revision: D76636927
@EddyLXJ EddyLXJ force-pushed the export-D76636927 branch from b3622b7 to d810c8b Compare June 18, 2025 01:24
@facebook-github-bot
Copy link
Contributor

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

EddyLXJ added a commit to EddyLXJ/torchrec that referenced this pull request Jun 18, 2025
Summary:
Pull Request resolved: pytorch#3106

# Change logs
1. add ZeroCollisionKeyValueEmbeddingBag emb lookup
2. address existing unit test missing for ebc ssd offloading
3. add new ut for kv zch embedding bag module

Embedding bag update is same as embedding update. Details can be found in D73567631

Reviewed By: duduyi2013

Differential Revision: D76636927
@EddyLXJ EddyLXJ force-pushed the export-D76636927 branch from d810c8b to 9373f9d Compare June 18, 2025 01:27
Summary:
Pull Request resolved: pytorch#3106

# Change logs
1. add ZeroCollisionKeyValueEmbeddingBag emb lookup
2. address existing unit test missing for ebc ssd offloading
3. add new ut for kv zch embedding bag module

Embedding bag update is same as embedding update. Details can be found in D73567631

Reviewed By: duduyi2013

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

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

@EddyLXJ EddyLXJ force-pushed the export-D76636927 branch from 9373f9d to 3bdca06 Compare June 18, 2025 01:35
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