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
Test Plan: Imported from OSS
Reviewed By: george-qi
Differential Revision: D33834916
Pulled By: anjali411
fbshipit-source-id: 11cec343e95e2ee188ab7576f26f64aa19317891
(cherry picked from commit f6e86f8a6b3924b67533183bd0b31e7ea4fcd2a9)
