Set correct device id on efficientzerotensors (#71611)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/71611
Fixes https://github.com/pytorch/pytorch/issues/71160 https://github.com/pytorch/pytorch/issues/69925 #69913
Test Plan: Imported from OSS
Reviewed By: VitalyFedyunin
Differential Revision: D33897543
Pulled By: anjali411
fbshipit-source-id: f1d8608c351876b8c2619da5ef891f74bad30ab5
(cherry picked from commit 643e666ea34aaf1584d478cd85025bf66d93cb21)