Skip to content

Commit c7a51c6

Browse files
mcremon-metafacebook-github-bot
authored andcommitted
Remove strictness in export calls
Summary: The new PT default has been to do non-strict export for a little while now. We have explicit strict because the PT compiler team preserved the current behavior at the time, but we should be able to remove it and match the default. It might also make more models go through. Differential Revision: D73859317
1 parent 072403b commit c7a51c6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/cadence/aot/compiler.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def prepare_and_convert_pt2(
9090
remove_decompositions(decomp_table, ops_to_keep)
9191
# Export with dynamo
9292
model_gm = (
93-
torch.export.export_for_training(model, inputs, strict=True)
93+
torch.export.export_for_training(model, inputs)
9494
.run_decompositions(decomp_table)
9595
.module()
9696
)
@@ -192,7 +192,7 @@ def export_program(
192192
torch._C._set_mkldnn_enabled(False)
193193

194194
# Export the model and return it.
195-
expo_program = export(model, inputs, strict=True)
195+
expo_program = export(model, inputs)
196196

197197
return expo_program
198198

0 commit comments

Comments
 (0)