pytorch/torch/_functorch
PyTorch MergeBot 361f42bc42 Revert "[compiled autograd] Compiled autograd configs in TLS (#137821)"
This reverts commit 9aba0b91c8.

Reverted https://github.com/pytorch/pytorch/pull/137821 on behalf of https://github.com/wdvr due to Reverting this for now, it is failing test_public_bindings in trunk ([comment](https://github.com/pytorch/pytorch/pull/137821#issuecomment-2417351788))
2024-10-16 16:38:29 +00:00
..
_aot_autograd Revert "[compiled autograd] Compiled autograd configs in TLS (#137821)" 2024-10-16 16:38:29 +00:00
__init__.py
aot_autograd.py fix test_lazy_module_kwargs (#137705) 2024-10-11 01:53:10 +00:00
apis.py
autograd_function.py Fix torch.library.register_vmap (#137306) 2024-10-04 03:46:35 +00:00
batch_norm_replacement.py
benchmark_utils.py
compile_utils.py Fix unintentional deduplication of returned tensors (#134726) 2024-09-04 23:42:56 +00:00
compilers.py
config.py Revert "Disallow FakeTensor.data_ptr access in eager mode (#137221)" 2024-10-07 21:46:13 +00:00
deprecated.py
eager_transforms.py
functional_call.py Allow optional positional arguments for torch.func.functional_call (#134643) 2024-09-12 15:22:06 +00:00
fx_minifier.py
make_functional.py
partitioners.py Remove two unused variables in _functorch/partitioners.py (#137998) 2024-10-16 10:58:31 +00:00
pyfunctorch.py
python_key.py
pytree_hacks.py
top_operators_github_usage.py
utils.py
vmap.py