torchrec
torchrec copied to clipboard
Add tests for group_tables in hbm cases
Summary: Adding a unit test for group_tables.
This way of testing isn't very reliable, since it just look at how the tables are grouped. Also it relies on ordering of the GroupedEmbeddingConfig, which is not ideal. But we should add that later.
Differential Revision: D50482054
This pull request was exported from Phabricator. Differential Revision: D50482054
This pull request was exported from Phabricator. Differential Revision: D50482054