pytorch/torch/export
Angela Yi de509abe1c [export] Dedup data-dependent errors based on stacktrace (#139540)
Summary:
Dedup the data-dependent errors based on the stacktrace it points to. Right now we just display every propagate-real-tensor log that shows up, but we actually can dedup them if they are due to the same piece of code (ex. there could multiple calls to a piece of code that does some data dependent computation).

This occurred when trying out draft export on the PT2I model zoo. For a specific model, previously we would get ~3k data dependent errors, but after deduping based on the stacktrace we now only get 4 errors.

Test Plan: CI

Differential Revision: D65374254

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139540
Approved by: https://github.com/pianpwk, https://github.com/zou3519
2024-11-05 18:16:05 +00:00
..
experimental Deprecate _preserve_ops and consolidate with decomp_table (#135080) 2024-09-15 17:01:58 +00:00
passes Fixing issue in move pass for copying Parameter (#138855) 2024-10-25 17:57:27 +00:00
__init__.py Replace torch.export default decomp table to be lazily populated (#137650) 2024-10-18 19:28:52 +00:00
_draft_export.py [export] Dedup data-dependent errors based on stacktrace (#139540) 2024-11-05 18:16:05 +00:00
_remove_auto_functionalized_pass.py Redesign custom op functionlaization for better re-inplace (#134409) 2024-09-04 17:08:58 +00:00
_remove_effect_tokens_pass.py [export][fx] More robust DCE pass (#132764) 2024-08-06 22:27: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 [export] Remove custom forward func in swap (#139126) 2024-10-30 16:50:57 +00:00
_trace.py Fix custom obj being input (#139209) 2024-11-04 18:24:29 +00:00
_tree_utils.py
_unlift.py Make test_torchbind.py training IR compatible (#138658) 2024-11-04 17:43:11 +00:00
custom_obj.py
decomp_utils.py Replace torch.export default decomp table to be lazily populated (#137650) 2024-10-18 19:28:52 +00:00
dynamic_shapes.py make DimHints compatible with Dims (#138490) 2024-10-22 07:43:48 +00:00
exported_program.py [export] Add support for symbool to make it usable for torch.cond (#138765) 2024-11-04 23:31:49 +00:00
graph_signature.py [export] Add support for symbool to make it usable for torch.cond (#138765) 2024-11-04 23:31:49 +00:00
unflatten.py [export] Add support for symbool to make it usable for torch.cond (#138765) 2024-11-04 23:31:49 +00:00