pytorch/torch/utils
Aaron Enye Shi dadb20a9d6 [Memory Snapshot][Viz] Add Allocator Settings Tab (#132518)
Summary: Since we are storing the allocator settings in the snapshot files for awhile now (since https://github.com/pytorch/pytorch/pull/119404), we can expose this to users with a new tab in the visualizer.

Test Plan:
Ran it locally:
![image](https://github.com/user-attachments/assets/5f79ccd0-fe1c-4e42-bb58-106d8f3cccd6)

Differential Revision: D60673548

Pulled By: aaronenyeshi

Pull Request resolved: https://github.com/pytorch/pytorch/pull/132518
Approved by: https://github.com/tianfengfrank, https://github.com/zdevito
2024-08-13 17:35:12 +00:00
..
_strobelight [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_sympy [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
backcompat
benchmark
bottleneck
data [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
hipify
jit
model_dump
tensorboard
viz [Memory Snapshot][Viz] Add Allocator Settings Tab (#132518) 2024-08-13 17:35:12 +00:00
__init__.py
_backport_slots.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_config_module.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_config_typing.pyi
_content_store.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_contextlib.py
_cpp_extension_versioner.py
_cxx_pytree.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_device.py
_exposed_in.py Revert "[BE] typing for decorators - _library/custom_ops (#131578)" 2024-07-28 03:29:32 +00:00
_foreach_utils.py
_freeze.py [BE] mypy: disallow untyped decorators (#131428) 2024-07-23 21:50:55 +00:00
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_ordered_set.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_python_dispatch.py return_and_correct_aliasing: skip dispatcher when swapping storage (#132524) 2024-08-06 00:44:35 +00:00
_pytree.py [pytree] Only import optree if it's used (#131478) 2024-07-24 00:10:49 +00:00
_stats.py
_thunk.py Refactor thunkify to return proper thunk abstraction (#132407) 2024-08-06 02:35:45 +00:00
_traceback.py
_triton.py Fix error handling in _triton.py (#132006) 2024-07-29 15:02:25 +00:00
_typing_utils.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py [Torch] Support meta device in checkpoint (#132684) 2024-08-06 20:45:50 +00:00
collect_env.py
cpp_backtrace.py
cpp_extension.py Loads .pyd instead of .so in MemPool test for windows (#132749) 2024-08-08 14:29:56 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py [NJT][flop counter] attention: if offsets are fake, use max seqlen (#132356) 2024-08-02 20:42:29 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
show_pickle.py
throughput_benchmark.py
weak.py