Simplify scaleup test to remove warn noise

pytorch / torchrec

Pytorch domain library for recommendation systems

https://pytorch.org/torchrec/

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

1.95k stars 441 forks source link

Simplify scaleup test to remove warn noise #2577

Closed damianr99 closed 4 days ago

damianr99 commented 5 days ago

Summary: D65774286 added additional logging to highlight when EmbeddingOffloadScaleupProposer searches only the cheapest of multiple sharding options provided. Update scaleup test so it does not unnecessarily enumerate multiple sharding options which triggers this warning.

Differential Revision: D66278106

facebook-github-bot commented 5 days ago

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

facebook-github-bot commented 5 days ago

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

facebook-github-bot commented 4 days ago

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

facebook-github-bot commented 4 days ago

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