From 0d750a3b89275da860d636843e9060977f6ac3fd Mon Sep 17 00:00:00 2001 From: "Yanan Cao (PyTorch)" Date: Thu, 19 Dec 2024 10:34:33 -0800 Subject: [PATCH] executorch/extension/android_test (#7394) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/7394 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67383536 --- extension/android_test/add_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extension/android_test/add_model.py b/extension/android_test/add_model.py index 5c7cf4770e2..b7ac3955ee3 100644 --- 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)