Brian Hirsh ed74dc054d add the option to disable functionalization in AOTDispatcher (#164577) ... I'm cleaning this PR up as a proper way of disabling functionalization via config in AOTDispatcher. I removed the non-functionalization related changes from the original version: (1) preventing proxy mode (and functionalization) from incorrectly decomposing CIA ops (Ed has a PR for it here: https://github.com/pytorch/pytorch/pull/164939) (2) preventing python-dispatcher-based decomps above autograd from running. I'm not doing this for now, will likely do it in a followup Pull Request resolved: https://github.com/pytorch/pytorch/pull/164577 Approved by: https://github.com/ezyang ghstack dependencies: #165372		2025-10-16 15:44:11 +00:00
..
_strobelight	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
_sympy	[torch/utils][Code Clean] Clean asserts in hipify/, jit/, model_dump and tensorboard of torch/utils (#165311)	2025-10-14 15:26:23 +00:00
backcompat
benchmark	[torch/utils][Code Clean] Clean asserts in benchmark/ and data/ in torch/utils/ (#165299)	2025-10-14 04:50:39 +00:00
data	[torch/utils][Code Clean] Clean asserts in benchmark/ and data/ in torch/utils/ (#165299)	2025-10-14 04:50:39 +00:00
hipify	[torch/utils][Code Clean] Clean asserts in hipify/, jit/, model_dump and tensorboard of torch/utils (#165311)	2025-10-14 15:26:23 +00:00
jit	[torch/utils][Code Clean] Clean asserts in hipify/, jit/, model_dump and tensorboard of torch/utils (#165311)	2025-10-14 15:26:23 +00:00
model_dump	[torch/utils][Code Clean] Clean asserts in hipify/, jit/, model_dump and tensorboard of torch/utils (#165311)	2025-10-14 15:26:23 +00:00
serialization
tensorboard	[torch/utils][Code Clean] Clean asserts in hipify/, jit/, model_dump and tensorboard of torch/utils (#165311)	2025-10-14 15:26:23 +00:00
viz	[2/N] More ruff SIM fixes (#165031)	2025-10-14 14:22:54 +00:00
__init__.py
_appending_byte_serializer.py	[2/N] Apply ruff UP035 check in torch files (#164054)	2025-09-29 03:35:32 +00:00
_config_module.py	Introduce joint_custom_pass callback (#164981)	2025-10-09 04:40:54 +00:00
_config_typing.pyi	Introduce joint_custom_pass callback (#164981)	2025-10-09 04:40:54 +00:00
_content_store.py
_contextlib.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
_cpp_embed_headers.py	[Fix] Adding missing f prefixes to formatted strings [2/N] (#164066)	2025-09-29 04:40:44 +00:00
_cpp_extension_versioner.py
_cxx_pytree.py	Pyrefly suppressions 7/n (#164913)	2025-10-08 07:27:17 +00:00
_debug_mode.py	[4/N] [DTensor device order] Support debugmode to show dtensor distribution transform path (#164821)	2025-10-11 09:44:54 +00:00
_device.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
_dtype_abbrs.py
_exposed_in.py	[2/N] Apply ruff UP035 check in torch files (#164054)	2025-09-29 03:35:32 +00:00
_filelock.py
_foreach_utils.py	[2/N] Apply ruff UP035 check in torch files (#164054)	2025-09-29 03:35:32 +00:00
_functools.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
_get_clean_triton.py
_helion.py
_import_utils.py
_mode_utils.py
_ordered_set.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
_python_dispatch.py	Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939)	2025-10-11 01:03:55 +00:00
_pytree.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
_stats.py	[2/N] Apply ruff UP035 check in torch files (#164054)	2025-09-29 03:35:32 +00:00
_thunk.py	[2/N] Apply ruff UP035 check in torch files (#164054)	2025-09-29 03:35:32 +00:00
_traceback.py
_triton.py	Workaround for mtia double init issue in has_triton (#162974)	2025-09-16 04:46:11 +00:00
_typing_utils.py
_zip.py
backend_registration.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
bundled_inputs.py	[2/N] Apply ruff UP035 check in torch files (#164054)	2025-09-29 03:35:32 +00:00
checkpoint.py	add the option to disable functionalization in AOTDispatcher (#164577)	2025-10-16 15:44:11 +00:00
collect_env.py	[2/N] Fix ruff warnings (#164460)	2025-10-04 03:40:32 +00:00
cpp_backtrace.py
cpp_extension.py	Fix #165125: Type "str" is not assignable to return type "None" (#165128)	2025-10-10 16:05:07 +00:00
deterministic.py
dlpack.py
file_baton.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
flop_counter.py	Pyrefly suppressions 7/n (#164913)	2025-10-08 07:27:17 +00:00
hooks.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py
throughput_benchmark.py
weak.py	Pyrefly suppressions 6/n (#164877)	2025-10-08 02:30:57 +00:00