pytorch/torch/_export
Tugsbayasgalan Manlaibaatar 8239ba4087 Fix various bugs in subclass input in export (#163770)
This adds basic support for subclass inputs in export (specifically for non-strict). I had to make fakify little more complicated which risks further divergence from dynamo fakification. But dynamo one is so complex, so i feel it is better to do this way. Also improved fake mode detection logic to recursively look into subclass inner tensors.

Differential Revision: [D83156489](https://our.internmc.facebook.com/intern/diff/D83156489)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/163770
Approved by: https://github.com/avikchaudhuri
2025-09-28 18:03:32 +00:00
..
db Move export_db to use new tracer, remove restriction on optional inputs (#162993) 2025-09-18 00:43:32 +00:00
pass_infra pyfmt lint more export files (#155783) 2025-06-12 23:04:11 +00:00
passes [aoti-fx] Add meta["val"] metadata (#161019) 2025-08-21 16:45:41 +00:00
serde [triton][export] serialization in internal path + unit tests (#162200) 2025-09-10 09:49:10 +00:00
__init__.py Move inductor.aot_compile to use new tracer (#163137) 2025-09-26 15:54:24 +00:00
config.py Improve fake tensor leakage detection in export by not relying on gc too much (#163516) 2025-09-22 22:04:24 +00:00
converter.py [BE][2/16] fix typos in torch/ (torch/_*/) (#156312) 2025-07-12 05:47:06 +00:00
error.py
non_strict_utils.py Fix various bugs in subclass input in export (#163770) 2025-09-28 18:03:32 +00:00
pass_base.py [BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551) 2025-06-25 06:16:06 +00:00
tools.py PEP585 update - torch/_export (#145138) 2025-01-19 18:48:35 +00:00
utils.py Don't skip register_dataclass unflatten in dynamo (#162557) 2025-09-17 16:53:02 +00:00
verifier.py Support vmap + custom autograd function/improve DTensor constructor inefficiency (#162240) 2025-09-11 17:42:41 +00:00
wrappers.py Support vmap + custom autograd function/improve DTensor constructor inefficiency (#162240) 2025-09-11 17:42:41 +00:00