From f7cb1ed4a3ec18d2e2d30521e49d9f57c7c80f89 Mon Sep 17 00:00:00 2001 From: "Yanan Cao (PyTorch)" Date: Thu, 19 Dec 2024 10:47:33 -0800 Subject: [PATCH] executorch/backends/xnnpack/test/tester (#7385) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/7385 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67381604 --- backends/xnnpack/test/tester/tester.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/xnnpack/test/tester/tester.py b/backends/xnnpack/test/tester/tester.py index c561f9f6617..1b6f03512bd 100644 --- a/backends/xnnpack/test/tester/tester.py +++ b/backends/xnnpack/test/tester/tester.py @@ -194,7 +194,7 @@ def run( inputs: Tuple[torch.Tensor], ) -> None: self.exported_program = export( - artifact, inputs, dynamic_shapes=self.dynamic_shapes + artifact, inputs, dynamic_shapes=self.dynamic_shapes, strict=True ) @property