pytorch
eda5abf5 - [quant][pt2e] Fix propagate_annotation after recent refactors (#102422)

Commit
1 year ago
[quant][pt2e] Fix propagate_annotation after recent refactors (#102422) Summary: Recently we changed the annotation from "target_dtype_info" to "quantization_annotation" and introduced QuantizationAnnotation API and SharedQuantizationSpec API for users to convey sharing between input/outputs, this PR updates the _propagate_annotation pass to accommadate the recent changes Test Plan: ``` buck2 test mode/opt caffe2/test:quantization_pt2e -- 'caffe2/test:quantization_pt2e' ``` Reviewed By: kimishpatel Differential Revision: D46153084 Pull Request resolved: https://github.com/pytorch/pytorch/pull/102422 Approved by: https://github.com/kimishpatel
Author
Committer
Parents
Loading