Skip to content

Commit 0d750a3

Browse files
gmagogsfmfacebook-github-bot
authored andcommitted
executorch/extension/android_test (pytorch#7394)
Summary: Pull Request resolved: pytorch#7394 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67383536
1 parent c337bef commit 0d750a3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

extension/android_test/add_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ def forward(self, x: torch.Tensor, y: torch.Tensor):
1313

1414

1515
# 1. torch.export: Defines the program with the ATen operator set.
16-
aten_dialect = export(Add(), (torch.ones(1), torch.ones(1)))
16+
aten_dialect = export(Add(), (torch.ones(1), torch.ones(1)), strict=True)
1717

1818
# 2. to_edge: Make optimizations for Edge devices
1919
edge_program = to_edge(aten_dialect)

0 commit comments

Comments
 (0)