[xnnpack] Reexport after quantize in aot_compiler (#7714)

pytorch · Jan 17, 2025 · 2f0518d · 2f0518d
1 parent 1b7b10e
commit 2f0518d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/xnnpack/aot_compiler.py b/examples/xnnpack/aot_compiler.py
@@ -92,6 +92,7 @@
         logging.info("Quantizing Model...")
         # TODO(T165162973): This pass shall eventually be folded into quantizer
         model = quantize(model, example_inputs)
+        ep = torch.export.export_for_training(model, example_inputs)
 
     edge = to_edge_transform_and_lower(
         ep,