saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/tools/autograd
History
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
deprecated.yaml
derivatives.yaml [MPS] cholesky implementation (#145701) 2025-01-27 01:53:03 +00:00
gen_annotated_fn_args.py
gen_autograd.py
gen_autograd_functions.py functional compiled autograd (#144707) 2025-01-27 05:20:56 +00:00
gen_inplace_or_view_type.py
gen_python_functions.py
gen_trace_type.py
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
load_derivatives.py
README.md

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.