pytorch
3f5f721a - Pass through allow-list from prepare_qat into propagate_qconfig_ to allow custom mapping and custom QAT module (#65119)

Commit
3 years ago
Pass through allow-list from prepare_qat into propagate_qconfig_ to allow custom mapping and custom QAT module (#65119) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65119 Pytorch Quantization: allow prepare_qat to include custom module by passing allow_list into the prepare_qat. When we are implementing custom module and custom mapping for Quantization Aware Training (QAT), we need to add the custom module to the mappings and to the allow_list during prepare_qat. The allow_list needs to be surfaced to the propagate_qconfig_. Test Plan: relying on general unit test Reviewed By: supriyar Differential Revision: D30982060 fbshipit-source-id: 1114115b6a3b853238d33d72b5cbaafc60f463e0
Author
Yuan Shangguan (June)
Parents
Loading