pytorch/torch/testing
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