pytorch/torch/export
Animesh Jain c2768d0f5a [export] Skip the check instead of disable (#164084)
Its unclear why we had disable in the first place. With
install_free_tensors, we are tracing into this hook. A better way would
be to place the tracer without any hook. For now, disable the checking
while dynamo is tracing.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/164084
Approved by: https://github.com/tugsbayasgalan
2025-09-29 01:44:44 +00:00
..
experimental [Torch Native] Add test for packaging weight (#158750) 2025-08-09 01:04:21 +00:00
passes [export] Move example inputs in move_to_device_pass (#162301) 2025-09-06 23:54:54 +00:00
pt2_archive [aoti] Save compute information (#163792) 2025-09-26 05:40:44 +00:00
__init__.py Fix inconsistent test and add new tracer as config (#162558) 2025-09-17 17:01:48 +00:00
_draft_export.py shape guards (#161178) 2025-09-08 22:44:09 +00:00
_leakage_detection_utils.py Prototype for building non-strict leak detector (#160456) 2025-09-03 19:21:27 +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
_swap.py shape guards (#161178) 2025-09-08 22:44:09 +00:00
_trace.py Fix preserve annotation with decomp (#163896) 2025-09-26 20:28:47 +00:00
_tree_utils.py PEP585 update - torch/export (#145165) 2025-01-19 20:56:55 +00:00
_unlift.py [export] Skip the check instead of disable (#164084) 2025-09-29 01:44:44 +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
custom_ops.py Support vmap + custom autograd function/improve DTensor constructor inefficiency (#162240) 2025-09-11 17:42:41 +00:00
decomp_utils.py Re-land exclude upsample_bilinear2d.vec and nearest2d.vec from default export decomposition table (#147153) 2025-02-19 23:03:29 +00:00
dynamic_shapes.py [Dynamic Shape][BE] trim _DimHint serialization (#163891) 2025-09-27 00:08:01 +00:00
exported_program.py testing infra and some fixes (#162183) 2025-09-10 20:48:12 +00:00
graph_signature.py Add support for param mutation under inference mode (#159661) 2025-08-14 03:34:04 +00:00
unflatten.py [export] Update unflattening dynamo.disable (#161306) 2025-08-27 00:27:16 +00:00