Revert "Remove device maps from TensorPipe for v1.7 release (#45353)" (#45762)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45762
This reverts commit 5211fb97ac4c246151f1286c78d63e0e317a8a4a.
Test Plan: Imported from OSS
Reviewed By: colesbury
Differential Revision: D24088231
Pulled By: mrshenli
fbshipit-source-id: b6ee15ec5ae137ea127bdc2db8e1842764bc01d4