pytorch

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

MPS: Add back the memory leak fixes. (#77964) Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/77964 Approved by: https://github.com/albanD