From 967a064503662fa25df0cf973982de234ded04a0 Mon Sep 17 00:00:00 2001 From: "Yanan Cao (PyTorch)" Date: Thu, 19 Dec 2024 10:47:30 -0800 Subject: [PATCH] executorch/build/packaging (#7386) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/7386 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67381635 --- build/packaging/smoke_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/build/packaging/smoke_test.py b/build/packaging/smoke_test.py index 1573e37bf5b..8f2bd08004f 100644 --- 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...")