pytorch

b2875268 - [bazel] use GPU machine and run GPU tests (#95721)

[bazel] use GPU machine and run GPU tests (#95721) Fixes #79354 Run bazel build and tests on GPU machine. Pull Request resolved: https://github.com/pytorch/pytorch/pull/95721 Approved by: https://github.com/huydhn