Zachary DeVito 3b3ed25109 Add a way to visualize memory snapshot traces (#90348) ... This adds a d3-based interactive visualization for exploring the memory allocation traces that the caching allocator can capture. This visualization code can also be attached to kineto trace information in the future to also provide visualization for the memory events captured there, which come with addition information about the graph. Pull Request resolved: https://github.com/pytorch/pytorch/pull/90348 Approved by: https://github.com/robieta		2022-12-10 02:45:11 +00:00
..
amp	Store autocast_gpu_dtype in custom_fwd and custom_bwd for BFloat16 autocast (#88029)	2022-10-31 22:45:26 +00:00
__init__.py	Fix exception causes all over the codebase (#90271)	2022-12-07 04:29:00 +00:00
_memory_viz.py	Add a way to visualize memory snapshot traces (#90348)	2022-12-10 02:45:11 +00:00
_sanitizer.py	Rework printing tensor aliases in CSAN error message (#85008)	2022-09-21 13:41:52 +00:00
_utils.py	Stream actually overrides __new__ so we need to patch it as well (#89592)	2022-11-29 21:43:23 +00:00
comm.py
error.py
graphs.py	[CUDA Graphs] Add option to dump a captured graph for debugging (#85519)	2022-12-06 22:03:05 +00:00
jiterator.py	Fix typos used in documents under torch directory (#88300)	2022-11-02 09:38:13 +00:00
memory.py	Add Pluggable CUDA allocator backend (#86786)	2022-11-23 17:54:36 +00:00
nccl.py
nvtx.py	Fix docstring inconsistencies: string -> str, boolean -> bool (#82410)	2022-07-28 21:29:57 +00:00
profiler.py	Add __all__ to torch.{autograd, fx, cuda} submodules (#85343)	2022-10-09 14:46:54 +00:00
random.py
sparse.py
streams.py