OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
047ae24e34
pytorch/torch/_functorch
History
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
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
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
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 Add support for tracing vmap in pre-dispatch export (#154650) 2025-08-20 19:31:07 +00:00