executorch/build/packaging (#7386)

Summary: Pull Request resolved: #7386 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67381635
pytorch · Dec 19, 2024 · 967a064 · 967a064
1 parent c337bef
commit 967a064
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build/packaging/smoke_test.py b/build/packaging/smoke_test.py
@@ -65,7 +65,7 @@ def export_linear_model() -> bytes:
 
     # Export the pytorch model and process for ExecuTorch.
     print("Exporting program...")
-    exported_program = export(LinearModel(), example_inputs)
+    exported_program = export(LinearModel(), example_inputs, strict=True)
     print("Lowering to edge...")
     edge_program = to_edge(exported_program)
     print("Creating ExecuTorch program...")