Syed Tousif Ahmed 1637a40796 Adds snapshot API for MemPools to get pool memory segments (#133601) ... Canonically, the snapshot API returns the entire memory state of the CUDACachingAllocator (using `get_all_blocks`). There is no API that can only return the memory state of a given pool. In this PR, we extend the functionality of snapshot API such that it can only return the memory addresses of an active pool. When snapshot API is called under a MemPoolContext, we only return the blocks that correspond to the pool id of the active pool. Part of https://github.com/pytorch/pytorch/issues/124807. Pull Request resolved: https://github.com/pytorch/pytorch/pull/133601 Approved by: https://github.com/ezyang		2024-10-29 01:01:47 +00:00
..
amp	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
__init__.py	Remove unused Python variables in torch/[b-z]* (#136963)	2024-10-19 16:45:22 +00:00
_gpu_trace.py
_memory_viz.py	Remove unused Python variables in torch/[b-z]* (#136963)	2024-10-19 16:45:22 +00:00
_sanitizer.py	Add proper handling for view and factory function for csan (#138236)	2024-10-18 14:04:18 +00:00
_utils.py
comm.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
error.py
gds.py	[Reland] Add wrappers for synchronous GPUDirect Storage APIs (#133489)	2024-08-15 17:11:52 +00:00
graphs.py	Remove unused Python variables in torch/[b-z]* (#136963)	2024-10-19 16:45:22 +00:00
jiterator.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
memory.py	Adds snapshot API for MemPools to get pool memory segments (#133601)	2024-10-29 01:01:47 +00:00
nccl.py	Flip default value for mypy disallow_untyped_defs [4/11] (#127841)	2024-06-08 18:36:48 +00:00
nvtx.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
profiler.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
random.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
sparse.py
streams.py	Use torch.Stream&torch.Event for Dynamo capature (#134850)	2024-10-02 14:15:33 +00:00
tunable.py	[ROCm] Tunableop record untuned (#128813)	2024-10-09 21:59:03 +00:00