Shivam Raikundalia a25a649e70 [Mem Snapshot] Add Metadata Field (#165490) ...

Summary:
The implementation adds the ability to:

Set custom metadata strings that will be attached to all subsequent allocations
Clear or change the metadata at any point
View the metadata in memory snapshots via _dump_snapshot()

Test Plan: Added test in test_cuda.py and check manually in snapshot to see that metadata was added.

Differential Revision: D84654933

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165490
Approved by: https://github.com/yushangdi

2025-10-17 23:46:02 +00:00

..

_acc

Add functions to setup PrivateUse1 as a python backend device. (#157859)

2025-10-01 21:32:59 +00:00

_dynamo

[5/N] Apply ruff UP035 rule (#164423)

2025-10-02 07:31:11 +00:00

_export

[serialization] Add pte file to archive (#162520)

2025-09-11 07:59:11 +00:00

__init__.pyi.in

[Mem Snapshot] Add Metadata Field (#165490)

2025-10-17 23:46:02 +00:00

_aoti.pyi

[AOTInductor] Add class declarations to torch._C._aoti interface file (#155128)

2025-06-17 00:10:57 +00:00

_autograd.pyi

[5/N] Apply ruff UP035 rule (#164423)

2025-10-02 07:31:11 +00:00

_cpu.pyi

[CPUInductor] Fix SVE256 detection (#146207)

2025-02-01 18:51:34 +00:00

_cudnn.pyi

Improve typing in torch/types.py (#145237)

2025-01-28 05:29:12 +00:00

_cusparselt.pyi

[sparse] Add cuSPARSELt as a backend (#128534)

2024-08-21 22:06:07 +00:00

_distributed_autograd.pyi

remove allow-untyped-defs for torch/_C/_distributed_autograd.pyi (#143369)

2024-12-17 18:09:28 +00:00

_distributed_c10d.pyi

Fix periodic debug tests failing due to FakeProcessGroup things (#165479)

2025-10-15 18:16:08 +00:00

_distributed_rpc_testing.pyi

Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)

2024-06-29 09:23:39 +00:00

_distributed_rpc.pyi

Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)

2024-06-29 09:23:39 +00:00

_functionalization.pyi

Reapply "Make functionalization ViewMeta serializable with pickle. (#143712)" (#163769)

2025-09-25 10:27:37 +00:00

_functions.pyi

PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102)

2025-01-18 20:47:12 +00:00

_functorch.pyi

torchdim Python port (#160236)

2025-09-21 03:01:04 +00:00

_instruction_counter.pyi

Add compile time instruction count metric (#133834)

2024-08-27 23:29:02 +00:00

_itt.pyi

Fix ITT unit-tests if PyTorch is compiled with USE_ITT=OFF (#86199)

2022-10-04 21:57:05 +00:00

_jit_tree_views.pyi

added stubs for jit tree views (#156504)

2025-06-25 06:15:17 +00:00

_lazy_ts_backend.pyi

Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)

2024-06-29 09:23:39 +00:00

_lazy.pyi

remove allow-untyped-defs for torch/_C/_lazy.pyi (#143370)

2024-12-17 17:18:10 +00:00

_monitor.pyi

[5/N] Apply ruff UP035 rule (#164423)

2025-10-02 07:31:11 +00:00

_nn.pyi.in

fix type hints for interpolation functions (#157202)

2025-07-09 03:11:37 +00:00

_nvtx.pyi

Inductor annotations (#130429)

2024-12-10 08:53:39 +00:00

_onnx.pyi

[1/N] [Caffe2] Remove caffe2_aten_fallback code (#128675)

2024-06-17 21:25:59 +00:00

_profiler.pyi

[5/N] Apply ruff UP035 rule (#164423)

2025-10-02 07:31:11 +00:00

_VariableFunctions.pyi.in

[BE][CI][Easy] Run lintrunner on generated .pyi stub files (#150732)

2025-05-27 14:58:02 +00:00

_verbose.pyi

[RFC] enable oneMKL&oneDNN on-demands verbose functinality (#63212)

2022-07-27 23:29:35 +00:00

build.bzl

Make all .pyi.in files exportable from torch/_C/ folder (#74962)

2022-03-31 12:54:14 +00:00

return_types.pyi.in

[BE][CI][Easy] Run lintrunner on generated .pyi stub files (#150732)

2025-05-27 14:58:02 +00:00