pytorch

dc2b2f09 - 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 (cherry picked from commit bde246fcc60372c0ce7ee16dd5e3dc7652a36867)