pytorch/torch/testing/_internal/optests
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
..
__init__.py
aot_autograd.py add the option to disable functionalization in AOTDispatcher (#164577) 2025-10-16 15:44:11 +00:00
autograd_registration.py
fake_tensor.py
generate_tests.py [6/N] Apply ruff UP035 rule (#164438) 2025-10-03 00:15:32 +00:00
make_fx.py [doc]: Small typos (#162982) 2025-09-16 17:42:19 +00:00