leslie-fang-intel 945a257277 [Quant][PT2E] Supported customized _EQUIVALENT_TYPES in Module Partition API (#102516) ... **Summary** `Module Partition API` can simplify the pattern match process in Quantization annotation. However, current implementation of `Module Partition API` has hardcoded `_EQUIVALENT_TYPES` 999bae0f54/torch/ao/quantization/_pt2e/graph_utils.py (L13-L20). So, PyTorch Extension Libraries such as [intel-extension-for-pytorch](https://github.com/intel/intel-extension-for-pytorch) can't use `Module Partition API` with customized `_EQUIVALENT_TYPES` . In this PR, we plan to enable customized `_EQUIVALENT_TYPES` by pass in parameter. **Test Plan** ``` python -m pytest test_graph_utils.py -k test_customized_equivalet_types_dict ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/102516 Approved by: https://github.com/jgong5, https://github.com/kimishpatel		2023-06-28 00:20:25 +00:00
..
ao_migration
bc
core	[ao] fixing quantized prelu workflow (#103455)	2023-06-23 16:45:40 +00:00
eager
fx	[PT2][Quant] Update op names for decomposed quantized lib (#103251)	2023-06-15 04:37:58 +00:00
jit
pt2e	[Quant][PT2E] Supported customized _EQUIVALENT_TYPES in Module Partition API (#102516)	2023-06-28 00:20:25 +00:00
serialized
__init__.py