From 8f85215e8f4d898f62299ba2b29d4939e239e7b3 Mon Sep 17 00:00:00 2001 From: Jerry Zhang Date: Thu, 13 Jul 2023 14:05:01 -0700 Subject: [PATCH] change internal code to only import from _quantize_pt2e (#7) Summary: X-link: https://github.com/pytorch/pytorch/pull/105162 Pull Request resolved: https://github.com/pytorch/executorch/pull/7 This is to make public api clear so that we can make implementation details change easier in the future Reviewed By: andrewor14 Differential Revision: D47445767 fbshipit-source-id: 74e31f15a03d47bbb0e7c9169ad3b7e467bf3b19 --- backends/xnnpack/test/test_xnnpack_utils.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/backends/xnnpack/test/test_xnnpack_utils.py b/backends/xnnpack/test/test_xnnpack_utils.py index acaedbf8cd7..7d261ee5544 100644 --- a/backends/xnnpack/test/test_xnnpack_utils.py +++ b/backends/xnnpack/test/test_xnnpack_utils.py @@ -42,11 +42,14 @@ QConfig, QConfigMapping, ) -from torch.ao.quantization._pt2e.quantizer import QNNPackQuantizer -from torch.ao.quantization._pt2e.quantizer.qnnpack_quantizer import ( + +from torch.ao.quantization._quantize_pt2e import ( + convert_pt2e, get_symmetric_quantization_config, + prepare_pt2e_quantizer, + QNNPackQuantizer, ) -from torch.ao.quantization._quantize_pt2e import convert_pt2e, prepare_pt2e_quantizer + from torch.ao.quantization.backend_config.executorch import ( get_executorch_backend_config, )