Edward Z. Yang e265b719bd Extract out prepare_aot_module_simplified for use in next PR (#158319) ... Also a small amount of extra code cleanup. Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/158319 Approved by: https://github.com/jingsh ghstack dependencies: #158149, #158150, #158173, #158176, #158213, #158251		2025-07-16 03:59:41 +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	Move functions from torch._functorch.aot_autograd that are not frontend functions to frontend_utils (#158251)	2025-07-16 03:59:41 +00:00
__init__.py
aot_autograd.py	Extract out prepare_aot_module_simplified for use in next PR (#158319)	2025-07-16 03:59:41 +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
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	[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)	2025-07-12 05:47:06 +00:00
compilers.py	[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)	2025-07-12 05:47:06 +00:00
config.py	Normalize placeholder names in AOTAutogradCache (#157916)	2025-07-14 17:45:11 +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	[Optimus][fp8_activation_quantization] Only log when there's some node to be quantized (#158129)	2025-07-15 19:22:26 +00:00
pyfunctorch.py	[dynamo] Guard serialization for FUNCTORCH_STACK_MATCH (#152616)	2025-05-05 18:05:56 +00:00
python_key.py
pytree_hacks.py
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