Skip to content

Commit eba96b2

Browse files
digantdesaiYIWENX14
authored andcommitted
[xnnpack] Reexport after quantize in aot_compiler (#7714)
1 parent 6cd70e7 commit eba96b2

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/xnnpack/aot_compiler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@
9292
logging.info("Quantizing Model...")
9393
# TODO(T165162973): This pass shall eventually be folded into quantizer
9494
model = quantize(model, example_inputs)
95+
ep = torch.export.export_for_training(model, example_inputs)
9596

9697
edge = to_edge_transform_and_lower(
9798
ep,

0 commit comments

Comments
 (0)