pytorch

baabb4cb - MPS: Add back the memory leak fixes. (#77964) (#78198)

MPS: Add back the memory leak fixes. (#77964) (#78198) Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/77964 Approved by: https://github.com/albanD Co-authored-by: Kulin Seth <kulinseth@gmail.com>