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 Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
_sympy Add sym_log2 (#137980) 2024-10-28 17:03:14 +00:00
backcompat
benchmark Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
bottleneck
data Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
hipify [ROCm] set hipblas workspace (#138791) 2024-10-29 01:37:55 +00:00
jit Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
model_dump Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
tensorboard Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
viz Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
__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 Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_ordered_set.py
_python_dispatch.py [BE]: Update Typeguard to TypeIs for better type inference (#133814) 2024-10-26 15:07:13 +00:00
_pytree.py [dynamo] support maxlen for collections.deque (#138194) 2024-10-30 10:08:02 +00:00
_stats.py
_thunk.py
_traceback.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
_triton.py Revert "[user triton] typing triton_kernel_wrap.py (#138230)" 2024-10-18 23:12:29 +00:00
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py Param fixes in docstring (#136097) 2024-09-21 18:56:34 +00:00
collect_env.py Update wmic command used in collect_env.py to its counterpart in powershell due to its deprecation (#138297) 2024-10-18 07:03:17 +00:00
cpp_backtrace.py
cpp_extension.py Enable Windows Arm64 (#133088) 2024-10-24 16:10:44 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py [BE]: Update mypy to 1.11.2 (#133816) 2024-09-16 19:44:11 +00:00