Skip to content

Commit f7e72ee

Browse files
authored
Enable strict=false in aot_compiler (#12872)
There were some issues with ic3/ci4 when strict=false, turning it on in aot compiler and seeing what throws.
1 parent 03f6bcc commit f7e72ee

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/xnnpack/aot_compiler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,14 +87,14 @@
8787

8888
model = model.eval()
8989
# pre-autograd export. eventually this will become torch.export
90-
ep = torch.export.export_for_training(model, example_inputs, strict=True)
90+
ep = torch.export.export_for_training(model, example_inputs, strict=False)
9191
model = ep.module()
9292

9393
if args.quantize:
9494
logging.info("Quantizing Model...")
9595
# TODO(T165162973): This pass shall eventually be folded into quantizer
9696
model = quantize(model, example_inputs, quant_type)
97-
ep = torch.export.export_for_training(model, example_inputs, strict=True)
97+
ep = torch.export.export_for_training(model, example_inputs, strict=False)
9898

9999
edge = to_edge_transform_and_lower(
100100
ep,

0 commit comments

Comments
 (0)