pytorch/tools/autograd/templates
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
..
ADInplaceOrViewType.cpp
annotated_fn_args.py.in
Functions.cpp
Functions.h
python_enum_tag.cpp
python_fft_functions.cpp
python_functions.cpp
python_functions.h
python_linalg_functions.cpp
python_nested_functions.cpp
python_nn_functions.cpp
python_return_types.cpp
python_return_types.h
python_sparse_functions.cpp
python_special_functions.cpp
python_torch_functions.cpp
python_variable_methods.cpp
TraceType.cpp
variable_factories.h
VariableType.cpp
VariableType.h
ViewFuncs.cpp
ViewFuncs.h