executorch/backends/apple/coreml/runtime/test

gmagogsfm · facebook-github-bot · commit 7f8196c0a6ed · 2024-12-18T16:30:04.000-08:00
Reviewed By: avikchaudhuri

Differential Revision: D67381348
diff --git a/backends/apple/coreml/runtime/test/export_stateful_model.py b/backends/apple/coreml/runtime/test/export_stateful_model.py
@@ -47,7 +47,7 @@ def main() -> None:
         torch.randn((1, embedding_dim)),
         torch.tensor([0]),
     )
-    exported_model = export(model, example_inputs)
+    exported_model = export(model, example_inputs, strict=True)
     edge_program_manager = exir.to_edge(exported_model)
     compile_specs = CoreMLBackend.generate_compile_specs(
         compute_precision=ct.precision.FLOAT16,

Original file line number	Diff line number	Diff line change
`@@ -47,7 +47,7 @@ def main() -> None:`
`47`	`47`	`torch.randn((1, embedding_dim)),`
`48`	`48`	`torch.tensor([0]),`
`49`	`49`	`)`
`50`		`- exported_model = export(model, example_inputs)`
	`50`	`+ exported_model = export(model, example_inputs, strict=True)`
`51`	`51`	`edge_program_manager = exir.to_edge(exported_model)`
`52`	`52`	`compile_specs = CoreMLBackend.generate_compile_specs(`
`53`	`53`	`compute_precision=ct.precision.FLOAT16,`