pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Yu, Guangye 15f1173e5d 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-06 02:22:18 +00:00

cpp

Removing conda references from PyTorch Docs (#152702 )

2025-05-20 20:33:28 +00:00

source

Add unified memory APIs for torch.accelerator (#152932 )

2025-08-06 02:22:18 +00:00

.gitignore

.gitignore for the docs folder

2019-10-08 12:18:30 -07:00

libtorch.rst

Add ROCm documentation to libtorch (C++) reST. (#136378 )