Kulin Seth 8df56afc20 Add support in Python API for the recommended max working set size. (#128289) ... Adds ways for users to request recommended max size for Metal on Mac. It plumbs through https://developer.apple.com/documentation/metal/mtldevice/2369280-recommendedmaxworkingsetsize?language=objc Can be used like ``` max_memory = torch.mps.recommended_max_memory() print ("Recommended Max Memory : ", (max_memory/(1024*1024*1024)), "GB") ``` Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/128289 Approved by: https://github.com/malfet		2024-06-12 16:03:57 +00:00
..
__init__.py	Add support in Python API for the recommended max working set size. (#128289)	2024-06-12 16:03:57 +00:00
event.py
profiler.py