pytorch

bde246fc - Speed up test_mps from 9min to 25s

Speed up test_mps from 9min to 25s Pull Request resolved: https://github.com/pytorch/pytorch/pull/78488 Approved by: https://github.com/kulinseth