Tugsbayasgalan Manlaibaatar 22ae059d32 AOTI util deprecated flow using the new tracer (#165582) ...

Reapply of https://github.com/pytorch/pytorch/pull/163260

AOTI utils expect free function sometimes so adjust export API to handle that, haven't seen any methods getting exported. Some AOTI flows also require we populate dynamo_flat_name_to_original_fqn so i just copy how it is done in eval_frame.py. I also cleaned up how we get rid of export_root and fixed some overcomplicated nn_module_stack handling in export code. The logic is simpler now thanks to @anijain2305 .

Pull Request resolved: https://github.com/pytorch/pytorch/pull/165582
Approved by: https://github.com/anijain2305

2025-10-19 15:52:16 +00:00

..

experimental

Add pyrefly suppressions (3/n) (#164588)

2025-10-03 22:03:03 +00:00

passes

Add pyrefly suppressions (3/n) (#164588)

2025-10-03 22:03:03 +00:00

pt2_archive

Enable all flake8-logging-format rules (#164655)

2025-10-19 00:59:28 +00:00

__init__.py

Enable all flake8-logging-format rules (#164655)

2025-10-19 00:59:28 +00:00

_draft_export.py

Add pyrefly suppressions (3/n) (#164588)

2025-10-03 22:03:03 +00:00

_leakage_detection_utils.py

More ruff SIM fixes (#164695)

2025-10-09 03:24:50 +00:00

_remove_auto_functionalized_pass.py

remove allow-untyped-defs from torch/export/_remove_auto_functionalized_pass.py (#144230)

2025-01-06 22:23:19 +00:00

_remove_effect_tokens_pass.py

[BE][1/16] fix typos in torch/ (#156311)

2025-07-09 11:02:22 +00:00

_safeguard.py

Flip default value for mypy disallow_untyped_defs [6/11] (#127843)

2024-06-08 18:49:29 +00:00

_swap.py

[2/N] Use "is" in python type comparison (#165142)

2025-10-10 15:36:44 +00:00

_trace.py

AOTI util deprecated flow using the new tracer (#165582)

2025-10-19 15:52:16 +00:00

_tree_utils.py

[5/N] Apply ruff UP035 rule (#164423)

2025-10-02 07:31:11 +00:00

_unlift.py

[2/N] More ruff SIM fixes (#165031)

2025-10-14 14:22:54 +00:00

_wrapper_utils.py

[reland] Make export._trace._WrapperModule work in strict mode (#146919) (#151264)

2025-04-15 18:35:34 +00:00

custom_obj.py

[Export] Introduce class_fqn into CustomObjArgument (#118158)

2024-01-25 18:44:25 +00:00

custom_ops.py

Support vmap + custom autograd function/improve DTensor constructor inefficiency (#162240)

2025-09-11 17:42:41 +00:00

decomp_utils.py

Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939)

2025-10-11 01:03:55 +00:00

dynamic_shapes.py

[1/N] Use "is" in python type comparison (#165037)

2025-10-10 12:36:50 +00:00

exported_program.py

[2/N] Fix ruff warnings (#164460)

2025-10-04 03:40:32 +00:00

graph_signature.py

Add support for param mutation under inference mode (#159661)

2025-08-14 03:34:04 +00:00

unflatten.py

Add pyrefly suppressions (3/n) (#164588)

2025-10-03 22:03:03 +00:00