Yu, Guangye 84f7e88aef Add unified memory APIs for torch.accelerator (#152932) ... # Motivation The following API will be put under torch.accelerator - empty_cache - max_memory_allocated - max_memory_reserved - memory_allocated - memory_reserved - memory_stats - reset_accumulated_memory_stats - reset_peak_memory_stats Pull Request resolved: https://github.com/pytorch/pytorch/pull/152932 Approved by: https://github.com/albanD ghstack dependencies: #138222		2025-08-08 17:41:22 +00:00
..
__init__.py	Add unified memory APIs for torch.accelerator (#152932)	2025-08-08 17:41:22 +00:00
_utils.py	[Accelerator][Chore] Use existing acc when raising an error (#150829)	2025-04-08 16:05:06 +00:00
memory.py	Add unified memory APIs for torch.accelerator (#152932)	2025-08-08 17:41:22 +00:00