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
Test Plan: Imported from OSS
Reviewed By: pbelevich, ngimel
Differential Revision: D33768645
Pulled By: anjali411
fbshipit-source-id: 66ce9907630b65a12c0775077147a7e72ff4cee4
(cherry picked from commit 3af98a4d707463a12f2b39bc839c5d7e51eec840)
