OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
0da8127f77
pytorch/torch/_functorch
History
Prajesh Praveen Anchalia 48e9ffc873 Unify on dynamo_compile as the overall wait counter (#150293) 
Summary:
dynamo_compile for the most part has been accounting for compile time except autotuning.

all_compilation_types had earlier been injected on fx_codegen_and_compile, which was incorrect.

Add autotuining to dynamo and deprcate all_compilation_types counter.

Differential Revision: D72145447

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150293
Approved by: https://github.com/masnesral, https://github.com/jamesjwu
2025-04-01 08:55:51 +00:00
..
_activation_checkpointing Do not depend on numpy during the import (#149683) 2025-03-21 08:14:57 +00:00
_aot_autograd Unify on dynamo_compile as the overall wait counter (#150293) 2025-04-01 08:55:51 +00:00
__init__.py
aot_autograd.py Improve attr mismatch msg (#149576) 2025-03-28 05:10:56 +00:00
apis.py [BE] typing for decorators (#144161) 2025-01-04 16:40:09 +00:00
autograd_function.py [AOTAutogradCache] Allow Custom Autograd functions behind a flag (#149751) 2025-03-24 21:12:11 +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 [Functorch] Fix devices Parameter Type in benchmark_utilization Function (#138774) 2024-10-25 19:25:18 +00:00
compile_utils.py
compilers.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
config.py [aotd] Config to guess_tangents_stride (#150035) 2025-03-28 13:54:19 +00:00
deprecated.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
eager_transforms.py Add torch.func.debug_unwrap (#146528) 2025-02-06 18:48:09 +00:00
functional_call.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
fx_minifier.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
make_functional.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
partitioners.py partitioner: ensure collectives saved by SAC that are actually unused in the bw are properly not saved (#149652) 2025-03-21 22:09:19 +00:00
pyfunctorch.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
python_key.py
pytree_hacks.py
top_operators_github_usage.py [BE]: Apply ruff PERF403 to use dict comprehensions more often (#149257) 2025-03-18 00:46:07 +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