pytorch
c3f0c77b - dbr quant support for custom leaf modules, part 3/x (#70349)

Commit
2 years ago
dbr quant support for custom leaf modules, part 3/x (#70349) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/70349 Makes sure that child modules of non traceable leaf modules do not participate in quantization swaps. This should feature complete the `non_traceable_module_class` feature. Test Plan: ``` python test/test_quantization.py TestQuantizeDBR.test_prepare_custom_config_dict_non_traceable_module_class ``` Reviewed By: VitalyFedyunin Differential Revision: D33296246 Pulled By: vkuzo fbshipit-source-id: 08287429c89ee6aa42d13ca3060a74679a478181
Author
Parents
Loading