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
..
_activation_checkpointing	[5/N] Apply ruff UP035 rule (#164423)	2025-10-02 07:31:11 +00:00
_aot_autograd	add the option to disable functionalization in AOTDispatcher (#164577)	2025-10-16 15:44:11 +00:00
__init__.py
aot_autograd.py	add the option to disable functionalization in AOTDispatcher (#164577)	2025-10-16 15:44:11 +00:00
apis.py	[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)	2025-06-25 06:16:06 +00:00
autograd_function.py	Add pyrefly suppressions (3/n) (#164588)	2025-10-03 22:03:03 +00:00
batch_norm_replacement.py
benchmark_utils.py	[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)	2025-06-25 06:16:06 +00:00
compile_utils.py	[5/N] Apply ruff UP035 rule (#164423)	2025-10-02 07:31:11 +00:00
compilers.py	[1/N] Use "is" in python type comparison (#165037)	2025-10-10 12:36:50 +00:00
config.py	add the option to disable functionalization in AOTDispatcher (#164577)	2025-10-16 15:44:11 +00:00
deprecated.py	[5/N] Apply ruff UP035 rule (#164423)	2025-10-02 07:31:11 +00:00
eager_transforms.py	[5/N] Apply ruff UP035 rule (#164423)	2025-10-02 07:31:11 +00:00
functional_call.py	[2/N] Use "is" in python type comparison (#165142)	2025-10-10 15:36:44 +00:00
fx_minifier.py	[5/N] Apply ruff UP035 rule (#164423)	2025-10-02 07:31:11 +00:00
make_functional.py	[2/N] Use "is" in python type comparison (#165142)	2025-10-10 15:36:44 +00:00
partitioners.py	add the option to disable functionalization in AOTDispatcher (#164577)	2025-10-16 15:44:11 +00:00
predispatch.py	Add support for tracing vmap in pre-dispatch export (#154650)	2025-08-20 19:31:07 +00:00
pyfunctorch.py	Add pyrefly suppressions (3/n) (#164588)	2025-10-03 22:03:03 +00:00
python_key.py
pytree_hacks.py
top_operators_github_usage.py	[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)	2025-06-25 06:16:06 +00:00
utils.py
vmap.py	[5/N] Apply ruff UP035 rule (#164423)	2025-10-02 07:31:11 +00:00