Conversation

EddyLXJ

Differential Revision: D76636927

@facebook-github-botfacebook--bot added the CLA SignedThis 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

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

@facebook-github-bot

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
@EddyLXJEddyLXJ force-pushed the export-D76636927 branch from 8447505 to b3622b7 Compare June 17, 2025 23:51
@facebook-github-bot

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
@EddyLXJEddyLXJ force-pushed the export-D76636927 branch from b3622b7 to d810c8b Compare June 18, 2025 01:24
@facebook-github-bot

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
@EddyLXJEddyLXJ 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

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

@EddyLXJEddyLXJ 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 . Already have an account? Sign in to comment
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.fb-exported
None yet

Successfully merging this pull request may close these issues.