pytorch
b66f1bc8 - fx quant: make forked subgraph rewriter preserve stack trace (#71858)

Commit
2 years ago
fx quant: make forked subgraph rewriter preserve stack trace (#71858) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71858 Makes the forked subgraph rewriter code path preserve stack traces. The strategy is pretty simple for now: 1. find any specified stack trace in pattern graph 2. if found, copy this stack trace to every node in replacement graph If more complicated logic is needed in the future, we can address it at a later time. Test Plan: ``` python test/test_quantization.py TestQuantizeFx.test_stack_trace_preserved_subgraph_rewriter ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D33791740 fbshipit-source-id: 38bb4885549a9f954278c6c14fa41f58f1d5f7b7 (cherry picked from commit 5cc32a87ce62ad9a1c8d2240cfe630cbf1cc838d)
Author
Committer
Parents
Loading