pytorch / torchrec

Pytorch domain library for recommendation systems

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

1.79k stars 381 forks source link

Closed TroyGarden closed 2 days ago

TroyGarden commented 2 days ago

Summary:

context

when we have the input tensors on the meta device, it calls the fake operator
however the device information is unintentionally missed so the output tensor is on the default device (cpu)
this is an incorrect behavior

Reviewed By: gnahzg, iamzainhuda

Differential Revision: D57077813

facebook-github-bot commented 2 days ago

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

facebook-github-bot commented 2 days ago

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

facebook-github-bot commented 2 days ago

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