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

pyfmt lint more torch/utils files (#155812)

2025-06-12 23:51:42 +00:00

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

Make record/storage alignment in torch.save configurable (#147788)

2025-03-06 12:04:46 +00:00

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

[BE] remove torch deploy - conditionals (#158288)

2025-07-29 17:40:49 +00:00

_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

Revert "Use the device interface for detecting Triton availability (#139171)"

2025-03-11 18:49:21 +00:00

_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

pyfmt lint more torch/utils files (#155812)

2025-06-12 23:51:42 +00:00

_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

[BE] Migrate dtype_abbrs into one location (#152229)

2025-04-28 03:52:47 +00:00

_exposed_in.py

[2/N] Apply ruff UP035 check in torch files (#164054)

2025-09-29 03:35:32 +00:00

_filelock.py

filelock: Make waitcounter variant to use (#139816)

2024-12-12 01:18:34 +00:00

_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

[BE]Enhance _get_clean_triton.py to auto-generate launch_params if missing (#154666)

2025-05-31 19:27:56 +00:00

_helion.py

Migrate from lru_cache to cache (#155613)

2025-06-11 19:44:18 +00:00

_import_utils.py

[BE] remove torch deploy - conditionals (#158288)

2025-07-29 17:40:49 +00:00

_mode_utils.py

pyfmt lint more torch/utils files (#155812)

2025-06-12 23:51:42 +00:00

_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

pyfmt lint more torch/utils files (#155812)

2025-06-12 23:51:42 +00:00

_triton.py

Workaround for mtia double init issue in has_triton (#162974)

2025-09-16 04:46:11 +00:00

_typing_utils.py

Revert "Fix type annotation of Linear.bias (#142326)"

2025-01-26 03:41:00 +00:00

_zip.py

pyfmt lint more torch/utils files (#155812)

2025-06-12 23:51:42 +00:00

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

Flip default value for mypy disallow_untyped_defs [9/11] (#127846)

2024-06-08 18:50:06 +00:00

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

Flip default value for mypy disallow_untyped_defs [9/11] (#127846)

2024-06-08 18:50:06 +00:00

dlpack.py

[DLPack] Add support for missing keyword-arguments. (#150218)

2025-07-20 00:46:20 +00:00

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

Flip default value for mypy disallow_untyped_defs [10/11] (#127847)

2024-06-08 18:50:06 +00:00

mobile_optimizer.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

model_zoo.py

…

module_tracker.py

[BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552)

2025-08-07 00:09:56 +00:00

show_pickle.py

Use typing.IO[bytes] instead of io.BytesIO in annotations (#144994)

2025-01-27 18:08:07 +00:00

throughput_benchmark.py

Flip default value for mypy disallow_untyped_defs [10/11] (#127847)

2024-06-08 18:50:06 +00:00

weak.py

Pyrefly suppressions 6/n (#164877)

2025-10-08 02:30:57 +00:00