Upstream generic device test patch. (#36321)
Summary:
So that XLA can run all tests by setting env `PYTORCH_TEST_PATH` instead of patching a diff. :)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36321
Differential Revision: D20946635
Pulled By: ailzhang
fbshipit-source-id: 55ab7db7fd93063ad495a0c23a903218a29625a4