forward fix embedding_lookup to include DTensor

pytorch / torchrec

Pytorch domain library for recommendation systems

https://pytorch.org/torchrec/

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

1.89k stars 416 forks source link

Closed iamzainhuda closed 3 months ago

iamzainhuda commented 3 months ago

Summary: TSIA, jobs failing because DTensor path does not exist

Reviewed By: TroyGarden

Differential Revision: D58983960

facebook-github-bot commented 3 months ago

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

facebook-github-bot commented 3 months ago

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