Add KeyValueParams - Githubissues

pytorch / torchrec

Pytorch domain library for recommendation systems

BSD 3-Clause "New" or "Revised" License

1.79k stars 381 forks source link

Closed henrylhtsang closed 1 week ago

henrylhtsang commented 1 week ago

Summary: Add KeyValueParams class, that are for params to go to SSD TBE.

Expectation:

pass to SSD TBE only when using EmbeddingComputeKernel.KEY_VALUE. This is important to make sure we can use a mixed of FUSED and KEY_VALUE tables.
need to be hashable

Differential Revision: D58892592

facebook-github-bot commented 1 week ago

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

facebook-github-bot commented 1 week ago

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

facebook-github-bot commented 1 week ago

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

facebook-github-bot commented 1 week ago

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