Move device guard from MultiTensorApply.cuh (#46664)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/46664
Test Plan: Imported from OSS
Reviewed By: anjali411
Differential Revision: D24453343
Pulled By: izdeby
fbshipit-source-id: b82a658af50ededc985195ed02dbf60e792c7a13