pytorch/torch/utils
Edward Z. Yang a6630bcf87 Profile guided optimization for automatic_dynamic (#139001)
Previously: https://github.com/pytorch/pytorch/pull/138052 but the implementation is done from scratch, so I open a new PR.

This implements the ability to save and load profiles of automatic dynamic decisions, so on subsequent runs we can directly make something automatically dynamic. Unlike the previous implementation, this cache is never enabled by default; instead, you have to specify a "job id" that says it's OK to share results. We will be able to automatically populate this id for internal MAST jobs but for generic OSS users you will have to explicitly opt into it.

Signed-off-by: Edward Z. Yang <ezyang@meta.com>

Differential Revision: [D65065497](https://our.internmc.facebook.com/intern/diff/D65065497)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/139001
Approved by: https://github.com/oulgen
2024-11-01 21:43:25 +00:00
..
_strobelight
_sympy Add sym_log2 (#137980) 2024-10-28 17:03:14 +00:00
backcompat
benchmark
bottleneck
data
hipify [ROCm] set hipblas workspace (#138791) 2024-10-29 01:37:55 +00:00
jit
model_dump
tensorboard
viz
__init__.py
_backport_slots.py
_config_module.py Profile guided optimization for automatic_dynamic (#139001) 2024-11-01 21:43:25 +00:00
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_extension_versioner.py Avoid file encoding issues when loading cpp extensions (#138565) 2024-10-28 14:06:34 +00:00
_cxx_pytree.py
_device.py
_exposed_in.py
_foreach_utils.py
_freeze.py
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_ordered_set.py
_python_dispatch.py
_pytree.py [dynamo] support maxlen for collections.deque (#138194) 2024-10-30 10:08:02 +00:00
_stats.py
_thunk.py
_traceback.py
_triton.py
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py
collect_env.py
cpp_backtrace.py
cpp_extension.py
deterministic.py
dlpack.py
file_baton.py
flop_counter.py
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py