Zachary DeVito 3e5a52cedd [memory snapshot] track context for segments (#106113) ... We want to display the stack for the original cudaMalloc that created a segment. Previously we could only report the last time the segment memory was used, or the record of the segment_alloc could appear in the list of allocator actions. This PR ensure regardless of whether we still have the segment_alloc action, the context for a segment is still available. The visualizer is updated to be able to incorporate this information. This PR adds a new field to Block. However the previous stacked cleanup PR removed a field of the same size, making the change to Block size-neutral. Pull Request resolved: https://github.com/pytorch/pytorch/pull/106113 Approved by: https://github.com/aaronenyeshi		2023-07-28 06:45:48 +00:00
..
_sympy	Prefer bound_sympy over sympy_interp (#105138)	2023-07-17 11:34:05 +00:00
backcompat
benchmark	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00
bottleneck	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00
data	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00
hipify	[BE] f-stringify torch/ and scripts (#105538)	2023-07-21 19:35:24 +00:00
jit	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00
model_dump	[BE]: Merge startswith calls - rule PIE810 (#96754)	2023-03-14 22:05:20 +00:00
tensorboard	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00
viz	[memory snapshot] track context for segments (#106113)	2023-07-28 06:45:48 +00:00
__init__.py	Add torch.utils to the docs page, remove dead code and fix docstrings (#105142)	2023-07-26 14:24:58 +00:00
_content_store.py	Make conversions from/to sparse semi-structured always @torch.compile-d (#105272)	2023-07-18 04:51:28 +00:00
_contextlib.py
_cpp_extension_versioner.py
_cuda_trace.py	Don't use f-strings in logging calls (1/X) (#98591)	2023-04-07 15:52:50 +00:00
_device.py	Guard on default device (#99551)	2023-04-20 17:02:59 +00:00
_foreach_utils.py	Merge and improve torch optim optimizer type stubs (#102593)	2023-07-26 11:56:42 +00:00
_freeze.py	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00
_mode_utils.py	Switch most Python RAII guard usages to context manager (#102642)	2023-06-01 16:28:37 +00:00
_python_dispatch.py	[dynamo] allow tensor subclass fakification in dynamo (#105308)	2023-07-18 17:28:04 +00:00
_pytree.py	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00
_stats.py
_traceback.py	[Reland] Update mypy to 1.4.1 (#105227)	2023-07-15 20:30:20 +00:00
_zip.py	[caffe2] Create deterministic zip archives (#102903)	2023-06-15 17:45:19 +00:00
backend_registration.py	Add torch.utils to the docs page, remove dead code and fix docstrings (#105142)	2023-07-26 14:24:58 +00:00
bundled_inputs.py	[BE] f-stringify torch/ and scripts (#105538)	2023-07-21 19:35:24 +00:00
checkpoint.py	[Composable] Use non-reentrant generator, remove reentrant (#105176)	2023-07-26 07:03:03 +00:00
collect_env.py	[BE]: Apply ruff PERF fixes to torch (#104917)	2023-07-11 20:45:21 +00:00
cpp_backtrace.py
cpp_extension.py	[BE] f-stringify torch/ and scripts (#105538)	2023-07-21 19:35:24 +00:00
dlpack.py	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00
file_baton.py	Fix typos under torch/utils directory (#97516)	2023-03-24 16:53:39 +00:00
flop_counter.py	[BE] f-stringify torch/ and scripts (#105538)	2023-07-21 19:35:24 +00:00
hooks.py	Added option to always call nn.Module global/non-global forward hooks (#104278)	2023-07-10 18:58:07 +00:00
mkldnn.py
mobile_optimizer.py	[BE] f-stringify torch/ and scripts (#105538)	2023-07-21 19:35:24 +00:00
model_zoo.py
show_pickle.py
throughput_benchmark.py	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00
weak.py	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00