OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
daff263062
pytorch/torch/_functorch
History
Aaron Gokaslan bbda22e648 [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 
Automated edits performed by FURB118. Operator is implemented in C and way faster when passed to another C method like sorted, max etc as a `key=`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154722
Approved by: https://github.com/jansel
2025-05-30 23:47:10 +00:00
..
_activation_checkpointing [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +00:00
_aot_autograd Revert "[internal] Expose additional metadata to compilation callbacks (#153596)" 2025-05-30 18:39:27 +00:00
__init__.py
aot_autograd.py [compile-time traces] Profile large missing gaps in compile time (#151256) 2025-05-13 14:44:51 +00:00
apis.py [BE] typing for decorators (#144161) 2025-01-04 16:40:09 +00:00
autograd_function.py Introduce unsafe way to mark functions as cacheable (#151603) 2025-04-21 17: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 [Functorch] Fix devices Parameter Type in benchmark_utilization Function (#138774) 2024-10-25 19:25:18 +00:00
compile_utils.py Don't CSE unbacked nodes (#154387) 2025-05-28 02:21:56 +00:00
compilers.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
config.py [aotd] Support saved tensors hooks in aot_autograd (#150032) 2025-05-22 14:09:38 +00:00
deprecated.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
eager_transforms.py [ca] disable ca for functorch grad and run all HOO tests (#154147) 2025-05-28 22:06:13 +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 [BE][Ez]: Optimize unnecessary lambda with operator (#154722) 2025-05-30 23:47:10 +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]: 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