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
..
amp
__init__.py	[BE] remove torch deploy - conditionals (#158288)	2025-07-29 17:40:49 +00:00
_gpu_trace.py
_memory_viz.py	[BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554)	2025-07-03 18:56:07 +00:00
_pin_memory_utils.py	[dcp] add new checkpoint staging to preserve storage sharing and support mutable state_dicts (#155192)	2025-06-19 02:04:21 +00:00
_sanitizer.py
_utils.py	simplify nvrtc discovery login in compile_kernel (#156674)	2025-06-24 08:55:40 +00:00
comm.py
error.py
gds.py	[BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554)	2025-07-03 18:56:07 +00:00
graphs.py	Dont't GC as often when collecting cudagraphs (#158193)	2025-07-24 21:37:11 +00:00
jiterator.py	[BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554)	2025-07-03 18:56:07 +00:00
memory.py	Add unified memory APIs for torch.accelerator (#152932)	2025-08-08 17:41:22 +00:00
nccl.py
nvtx.py
profiler.py
random.py
sparse.py
streams.py	Support stream capture of event record and wait nodes in cuda graphs (#155372)	2025-06-17 21:44:51 +00:00
tunable.py	[BE][1/16] fix typos in torch/ (#156311)	2025-07-09 11:02:22 +00:00