pytorch/torch
Way Wang 4a94591321 filter out alloc-free pairs from trace plot (#165752)
Summary:
When dealing with a large memory trace, the resulting plot can be challenging to interpret and analyze.
This commit introduces a feature that enables filtering of allocations that have already been freed, providing a more focused view.
The remaining events in the plot often warrant closer examination, as they may be indicative of potential out-of-memory (OOM) issues.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165752
Approved by: https://github.com/zdevito
2025-10-29 12:44:54 +00:00
..
_awaits
_C Introduce a new API torch.xpu.set_per_process_memory_fraction (#165510) 2025-10-29 03:24:52 +00:00
_C_flatbuffer
_custom_op Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_decomp Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_dispatch Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_dynamo Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
_export [pytree] add treespec_{leaf,tuple,dict} functions for args_spec modification (#160843) 2025-10-29 09:16:24 +00:00
_functorch Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
_higher_order_ops Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
_inductor Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
_lazy Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_library Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_logging Clean up unused Pyrefly suppressions (#166178) 2025-10-25 05:32:21 +00:00
_numpy Enable PLW0127 in ruff (#165851) 2025-10-21 03:30:57 +00:00
_prims Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_prims_common Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_refs Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_strobelight Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_subclasses Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
_vendor
accelerator Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
amp Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
ao Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
autograd Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
backends Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
compiler Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
contrib
cpu Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
csrc [User-streams] Make torch.Event weakref compatible (#164522) 2025-10-29 04:57:23 +00:00
cuda filter out alloc-free pairs from trace plot (#165752) 2025-10-29 12:44:54 +00:00
distributed Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
distributions Fix pyrelfy ignore syntax in distributions and ao (#166248) 2025-10-26 22:13:48 +00:00
export Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
fft
func
futures
fx Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
headeronly Remove likely unnecessary _EXPAND trick for non-windows in HIDDEN_NAMESPACE_BEGIN (#166203) 2025-10-25 04:44:07 +00:00
jit Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
legacy
lib [2/N] Mark unused parameters in C++ code (#165121) 2025-10-15 03:04:39 +00:00
linalg
masked Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
monitor
mps
mtia Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
multiprocessing Replace NUMA inheritance approach (#166026) 2025-10-29 03:58:44 +00:00
nativert [triton][nativert] Add num_cpu_threads for triton-cpu (#166255) 2025-10-28 08:40:04 +00:00
nested Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
nn Fix existing Pyrefly errors (#166439) 2025-10-29 02:08:02 +00:00
numa Replace NUMA inheritance approach (#166026) 2025-10-29 03:58:44 +00:00
onnx Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
optim Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
package Clean up unused Pyrefly suppressions (#166178) 2025-10-25 05:32:21 +00:00
profiler Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
quantization [RFC] Add pyrefly to lintrunner (#165179) 2025-10-16 20:07:09 +00:00
signal Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
sparse Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
special
testing Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
utils Revert "[1/N] Remove unused loop variables (#166258)" 2025-10-29 11:10:37 +00:00
xpu Introduce a new API torch.xpu.set_per_process_memory_fraction (#165510) 2025-10-29 03:24:52 +00:00
__config__.py
__future__.py
__init__.py Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_appdirs.py
_classes.py
_compile.py
_custom_ops.py
_environment.py
_guards.py Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
_jit_internal.py Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_linalg_utils.py
_lobpcg.py Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_lowrank.py
_meta_registrations.py [Inductor][Triton][FP8] Support deepseek-style scaling in Inductor (#164404) 2025-10-28 03:38:54 +00:00
_namedtensor_internals.py
_ops.py Fix pyrefly ignore syntax (#166438) 2025-10-29 00:02:21 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py
_tensor_str.py Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_tensor.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_thread_safe_fork.py
_torch_docs.py Clarrifying input output angle unit in the docs for trigonometric fun… (#161248) 2025-10-18 11:53:48 +00:00
_utils_internal.py Fix pyrefly error syntax (2/n) (#166448) 2025-10-29 00:36:40 +00:00
_utils.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
CMakeLists.txt [ROCm] Use a ROCm version string without hash. (#166336) 2025-10-28 03:53:55 +00:00
custom_class_detail.h Mark unused parameters in C++ code (#164912) 2025-10-09 06:23:25 +00:00
custom_class.h Mark unused parameters in C++ code (#164912) 2025-10-09 06:23:25 +00:00
extension.h
functional.py Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
header_only_apis.txt Move toUnderlying to headeronly (#165694) 2025-10-22 05:31:16 +00:00
hub.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
library.h Mark unused parameters in C++ code (#164912) 2025-10-09 06:23:25 +00:00
library.py Clean up unused Pyrefly suppressions (#166178) 2025-10-25 05:32:21 +00:00
overrides.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
py.typed
quasirandom.py
random.py Fix flake8 B028 warnings (#166224) 2025-10-26 06:18:55 +00:00
return_types.py
script.h
serialization.py Document limitations of weights_only in SECURITY.md and torch.load doc (#165645) 2025-10-27 18:20:50 +00:00
storage.py Fix pyrefly ignores 1/n (#166239) 2025-10-26 00:44:10 +00:00
torch_version.py
types.py Enable PLC0414 on ruff (#165828) 2025-10-22 04:56:52 +00:00
version.py.tpl