executorch/extension/android_test (pytorch#7394)

Summary: Pull Request resolved: pytorch#7394 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67383536
gmagogsfm · Dec 19, 2024 · 0d750a3 · 0d750a3
1 parent c337bef
commit 0d750a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extension/android_test/add_model.py b/extension/android_test/add_model.py
@@ -13,7 +13,7 @@ def forward(self, x: torch.Tensor, y: torch.Tensor):
 
 
 # 1. torch.export: Defines the program with the ATen operator set.
-aten_dialect = export(Add(), (torch.ones(1), torch.ones(1)))
+aten_dialect = export(Add(), (torch.ones(1), torch.ones(1)), strict=True)
 
 # 2. to_edge: Make optimizations for Edge devices
 edge_program = to_edge(aten_dialect)