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