change internal code to only import from _quantize_pt2e (#7)

jerryzh168 · facebook-github-bot · commit c32f9b44e3a2 · 2023-07-13T20:39:40.000-07:00
Summary: X-link: pytorch/pytorch#105162 Pull Request resolved: #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: e6887b92e0fe9112c9a106e6ec31a24d080e4028
diff --git 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,
 )

Original file line number	Diff line number	Diff line change
`@@ -42,11 +42,14 @@`
`42`	`42`	`QConfig,`
`43`	`43`	`QConfigMapping,`
`44`	`44`	`)`
`45`		`-from torch.ao.quantization._pt2e.quantizer import QNNPackQuantizer`
`46`		`-from torch.ao.quantization._pt2e.quantizer.qnnpack_quantizer import (`
	`45`	`+`
	`46`	`+from torch.ao.quantization._quantize_pt2e import (`
	`47`	`+ convert_pt2e,`
`47`	`48`	`get_symmetric_quantization_config,`
	`49`	`+ prepare_pt2e_quantizer,`
	`50`	`+ QNNPackQuantizer,`
`48`	`51`	`)`
`49`		`-from torch.ao.quantization._quantize_pt2e import convert_pt2e, prepare_pt2e_quantizer`
	`52`	`+`
`50`	`53`	`from torch.ao.quantization.backend_config.executorch import (`
`51`	`54`	`get_executorch_backend_config,`
`52`	`55`	`)`