Make sure XNNPACKQuantizer works with the pre_dispatch=True

jerryzh168 · facebook-github-bot · commit 6b350bd4d916 · 2023-08-24T21:55:35.000-07:00
Summary: X-link: pytorch/pytorch#107872 att Reviewed By: andrewor14, tugsbayasgalan Differential Revision: D48415977 fbshipit-source-id: fc8c8bfc7ba64b53e0f3dee517c4d95462735d9c
diff --git a/backends/xnnpack/test/test_xnnpack_utils.py b/backends/xnnpack/test/test_xnnpack_utils.py
@@ -307,7 +307,8 @@ def quantize_and_test_model_with_quantizer(
         module.eval()
         # program capture
         m = torch._export.capture_pre_autograd_graph(
-            module, example_inputs, decomp_table=_default_decomposition_table()
+            module,
+            example_inputs,
         )
 
         quantizer = XNNPACKQuantizer()

Original file line number	Diff line number	Diff line change
`@@ -307,7 +307,8 @@ def quantize_and_test_model_with_quantizer(`
`307`	`307`	`module.eval()`
`308`	`308`	`# program capture`
`309`	`309`	`m = torch._export.capture_pre_autograd_graph(`
`310`		`- module, example_inputs, decomp_table=_default_decomposition_table()`
	`310`	`+ module,`
	`311`	`+ example_inputs,`
`311`	`312`	`)`
`312`	`313`
`313`	`314`	`quantizer = XNNPACKQuantizer()`