rzou aa2d54148d Add AOTDispatcher config to set backward autocast behavior (#156356) ...

This PR adds a new config `backward_pass_autocast`, to set the backward autocast
behavior. It does not change the existing behavior.

The reason why we need this is that torch.compile acquires a forward and
backward graph at the time of the forward pass. This means that
implemented naively, if there are any context managers active outside
the call to torch.compile, the backward graph will also get the
behaviors from those context managers. This PR gives users a way to
tweak the autocast behavior of the backward pass.

Please see torch._functorch.config for the options to the
`backward_pass_autocast` config.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156356
Approved by: https://github.com/bdhirsh
ghstack dependencies: #155354

2025-06-27 14:58:58 +00:00

..

_activation_checkpointing

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

_aot_autograd

Add AOTDispatcher config to set backward autocast behavior (#156356)

2025-06-27 14:58:58 +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

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

apis.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

autograd_function.py

Typo fixes for "overridden" in comments and function names (#155944)

2025-06-14 03:37:38 +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

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

compile_utils.py

[PT2][partitioners] raise getitems in partitioners to allow earlier release of buffers (#155809)

2025-06-21 19:57:21 +00:00

compilers.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

config.py

Add AOTDispatcher config to set backward autocast behavior (#156356)

2025-06-27 14:58:58 +00:00

deprecated.py

PEP585 update - torch/_functorch (#145139)

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

eager_transforms.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

functional_call.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

fx_minifier.py

PEP585 update - torch/_functorch (#145139)

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

make_functional.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

partitioners.py

[aotd] Support mutations of the same input in fw and bw (#155354)

2025-06-26 14:05:54 +00:00

pyfunctorch.py

[dynamo] Guard serialization for FUNCTORCH_STACK_MATCH (#152616)

2025-05-05 18:05:56 +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][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

utils.py

PEP585 update - torch/_functorch (#145139)

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

vmap.py

[dynamo] Support functools.partial variables through inspect.signature (#146339)

2025-02-04 04:39:39 +00:00