Skip to content

Commit 666b50d

Browse files
EddyLXJfacebook-github-bot
authored andcommitted
torchrec support on kvzch ebc emb lookup module (#3106)
Summary: # 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, kausv Differential Revision: D76636927
1 parent aaa8d25 commit 666b50d

File tree

4 files changed

+914
-56
lines changed

4 files changed

+914
-56
lines changed

0 commit comments

Comments
 (0)