pytorch/test/quantization
Jerry Zhang 793b17ebcb Add numeric_debugger top level APIs (#130643)
Summary:
Add three top level APIs for numeric debugger in pt2e flow that can log intermediate output in the model
and calculate summary for metric comparisons between nodes in two graphs

* `prepare_for_propagation_comparison`
* `extract_results_from_loggers`
* `compare_results`

Test Plan:
python test/test_quantization.py -k test_prepare_for_propagation_comparison
python test/test_quantization.py -k test_extract_results_from_loggers

Reviewers:

Subscribers:

Tasks:

Tags:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/130643
Approved by: https://github.com/dulinriley, https://github.com/tarun292
2024-07-18 20:54:18 +00:00
..
ao_migration
bc
core Fix for HistogramObserver (#129387) 2024-07-02 15:41:44 +00:00
eager Support min/max carry over for eager mode from_float method (#127309) 2024-05-29 19:33:26 +00:00
fx
jit
pt2e Add numeric_debugger top level APIs (#130643) 2024-07-18 20:54:18 +00:00
serialized
__init__.py