pytorch/torch/_export
angelayi 28bcd9eb30 [aoti] Initial Metal support (#153959)
An example generated file: P1816629015

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153959
Approved by: https://github.com/malfet, https://github.com/desertfire
ghstack dependencies: #153964
2025-05-21 21:55:59 +00:00
..
db PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
pass_infra PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
passes [export] Remove unused constants (#153800) 2025-05-20 03:15:27 +00:00
serde GPU lowering uses aoti_call_delegate (#153282) 2025-05-13 23:23:27 +00:00
__init__.py [aoti] Initial Metal support (#153959) 2025-05-21 21:55:59 +00:00
converter.py Revert "Improve torch.ops typing (#153558)" 2025-05-19 23:32:36 +00:00
error.py Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
non_strict_utils.py [export] support functools.partial forward (non-strict) (#153408) 2025-05-13 23:30:13 +00:00
pass_base.py BE: Swap functorch --> torch._higher_order_ops (#152620) 2025-05-01 21:53:23 +00:00
tools.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
utils.py [export] Dynamo symint support (#152677) 2025-05-16 07:51:50 +00:00
verifier.py Expand allowed_getattr_types_for_subgm to torch.Tensor (#150867) 2025-04-09 11:01:45 +00:00
wrappers.py Support subclass constructor capturing in export (#147014) 2025-03-16 18:19:19 +00:00