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

..

_internal

add the option to disable functionalization in AOTDispatcher (#164577)

2025-10-16 15:44:11 +00:00

__init__.py

Add pyrefly suppressions 2/n (#164513)

2025-10-03 02:46:13 +00:00

_comparison.py

Pyrefly suppressions 6/n (#164877)

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

_creation.py

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

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

_utils.py

Fix code descriptions in the test package. (#148145)

2025-03-04 19:14:41 +00:00