pytorch/torch/_export
Avik Chaudhuri 51eacea8c4 graph module retracing without preserving MCS (#143676)
Retracing while preserving module call signatures used to be a problem because graph modules don't have submodules at given paths. This led to a number of failing retracebility tests. By not trying to wrap modules with export tracepoints we can pass most of these tests; the only exception is where you do module swapping on retraced programs, which is still not possible.

Differential Revision: [D67539304](https://our.internmc.facebook.com/intern/diff/D67539304/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143676
Approved by: https://github.com/zhxchen17, https://github.com/tugsbayasgalan
ghstack dependencies: #143664
2024-12-21 07:57:43 +00:00
..
db Improve cond error messaging (#143595) 2024-12-20 01:19:20 +00:00
pass_infra
passes remove allow-untyped-defs for torch/_export/passes/remove_runtime_assertions.py (#143435) 2024-12-18 03:05:20 +00:00
serde Add float8 support in serde schema (#143343) 2024-12-18 05:07:21 +00:00
__init__.py [reland] Kill capture_pre_autograd_graph API (#143426) 2024-12-18 12:07:09 +00:00
converter.py [ts converter] use Dim.AUTO for ts -> export converter (#138273) 2024-12-20 07:48:24 +00:00
error.py
non_strict_utils.py support slicing with symints in non-strict (#143217) 2024-12-14 10:27:45 +00:00
pass_base.py Support symbolic builtin round in export (#139549) 2024-11-07 02:49:44 +00:00
tools.py
utils.py [export] add is_exporting flag (#142425) 2024-12-18 21:36:28 +00:00
verifier.py Add truediv support in export serializer (#136364) 2024-12-05 17:33:33 +00:00
wrappers.py graph module retracing without preserving MCS (#143676) 2024-12-21 07:57:43 +00:00