Yidi Wu 8f6dbc0ba8 [scan] create fw and bw graphs via partitioning (#162754) ...

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162754
Approved by: https://github.com/zou3519
ghstack dependencies: #161557, #161664, #161808, #162025, #161732

2025-09-27 18:13:15 +00:00

..

_activation_checkpointing

Enable logging for absolute memory estimation (#158799)

2025-09-22 18:36:49 +00:00

_aot_autograd

Update ruff to 0.13.1 (#163744)

2025-09-26 10:12:21 +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

Allow aot_module_simplified to return a serializable output (#162527)

2025-09-16 15:22:05 +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

[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

[Precompile] [RFC] Implement aot_compile_module (#162171)

2025-09-14 23:32:28 +00:00

deprecated.py

PEP585 update - torch/_functorch (#145139)

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

eager_transforms.py

[doc]: Small typos (#162982)

2025-09-16 17:42:19 +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

[scan] create fw and bw graphs via partitioning (#162754)

2025-09-27 18:13:15 +00:00

predispatch.py

Add support for tracing vmap in pre-dispatch export (#154650)

2025-08-20 19:31:07 +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

Add support for tracing vmap in pre-dispatch export (#154650)

2025-08-20 19:31:07 +00:00