pytorch/tools/autograd
rzou ea141d8134 functional compiled autograd (#144707)
This PR squashes together the following commits:

https://github.com/pytorch/pytorch/pull/144115
https://github.com/pytorch/pytorch/pull/143417
https://github.com/pytorch/pytorch/pull/143405
https://github.com/pytorch/pytorch/pull/143387
https://github.com/pytorch/pytorch/pull/143304
https://github.com/pytorch/pytorch/pull/143296

This is a refactor of compiled autograd to use "functional autograd". The end goal is that it gets compiled autograd's initial capture to stop specializing on Tensor metadata, therefore allowing compiled autograd to better handle Tensor subclasses.

For more information, please read the commit messages for each PR.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/144707
Approved by: https://github.com/bdhirsh, https://github.com/xmfan, https://github.com/jansel
2025-01-27 05:20:56 +00:00
..
templates functional compiled autograd (#144707) 2025-01-27 05:20:56 +00:00
__init__.py
BUILD.bazel
build.bzl
context.py [BE] Format .ci/ / .github/ / benchmarks/ / functorch/ / tools/ / torchgen/ with ruff format (#132577) 2024-10-11 18:30:26 +00:00
deprecated.yaml
derivatives.yaml [MPS] cholesky implementation (#145701) 2025-01-27 01:53:03 +00:00
gen_annotated_fn_args.py [3/N] Apply py39 ruff fixes (#142115) 2024-12-11 17:50:10 +00:00
gen_autograd.py [BE] Format .ci/ / .github/ / benchmarks/ / functorch/ / tools/ / torchgen/ with ruff format (#132577) 2024-10-11 18:30:26 +00:00
gen_autograd_functions.py functional compiled autograd (#144707) 2025-01-27 05:20:56 +00:00
gen_inplace_or_view_type.py Fix unused Python variables outside torch/ and test/ (#136359) 2024-12-11 17:10:23 +00:00
gen_python_functions.py [3/N] Apply py39 ruff fixes (#142115) 2024-12-11 17:50:10 +00:00
gen_trace_type.py [3/N] Apply py39 ruff fixes (#142115) 2024-12-11 17:50:10 +00:00
gen_variable_factories.py
gen_variable_type.py Fix unbind_copy and add its decomposition (#134319) 2025-01-17 18:21:22 +00:00
gen_view_funcs.py Fix Wextra-semi warnings (#139000) 2024-10-28 21:48:51 +00:00
load_derivatives.py [3/N] Apply py39 ruff fixes (#142115) 2024-12-11 17:50:10 +00:00
README.md

If you add a file to this directory, you MUST update torch/CMakeLists.txt and add the file as a dependency to the add_custom_command call.