pytorch

138c944a - [ROCm] default tests use 1 GPU, distributed tests use 2 GPUs

[ROCm] default tests use 1 GPU, distributed tests use 2 GPUs Fixes #76497. Pull Request resolved: https://github.com/pytorch/pytorch/pull/76499 Approved by: https://github.com/janeyx99