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

..

_activation_checkpointing

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

_aot_autograd

functional compiled autograd (#144707)

2025-01-27 05:20:56 +00:00

__init__.py

Revert "Replace _dynamo.config with an object instead of module (#96455)"

2023-06-05 07:06:14 +00:00

aot_autograd.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

apis.py

[BE] typing for decorators (#144161)

2025-01-04 16:40:09 +00:00

autograd_function.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

batch_norm_replacement.py

remove allow-untyped-defs for torch/_functorch/batch_norm_replacement.py (#143438)

2024-12-18 09:01:06 +00:00

benchmark_utils.py

[Functorch] Fix devices Parameter Type in benchmark_utilization Function (#138774)

2024-10-25 19:25:18 +00:00

compile_utils.py

Fix unintentional deduplication of returned tensors (#134726)

2024-09-04 23:42:56 +00:00

compilers.py

Remove unused Python variables in torch/[_-a]* (#133492)

2024-12-12 17:39:14 +00:00

config.py

[reland][export] don't decompose custom triton op when exporting (#144284)

2025-01-11 01:34:35 +00:00

deprecated.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

eager_transforms.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

functional_call.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

fx_minifier.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

make_functional.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

partitioners.py

[BE]: Simplify set add with set update (#145152)

2025-01-23 20:18:13 +00:00

pyfunctorch.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

python_key.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00

pytree_hacks.py

[BE] wrap deprecated function/class with typing_extensions.deprecated (#127689)

2024-06-02 12:30:43 +00:00

top_operators_github_usage.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00

utils.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

vmap.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00