torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Integrate heterogenous auto planner to TGIF DI

Open gnahzg opened this issue 1 year ago • 1 comments

Summary: As titled

Integrate TorchRec autoplanner for uneven sharding through option use_trec_hetero_auto_planner. The autoplanner would only be used when user explicitly set so.

Continue testing would be done before we fully migrate to use TorchRec Hetero Autoplanner. Aiming for timeline end of H1 2024

Differential Revision: D56203531

gnahzg avatar May 10 '24 15:05 gnahzg

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

facebook-github-bot avatar May 10 '24 15:05 facebook-github-bot

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

facebook-github-bot avatar May 13 '24 19:05 facebook-github-bot