From 484db32c6ab0cab196a278e21a5e42bc4d671262 Mon Sep 17 00:00:00 2001
From: "Yanan Cao (PyTorch)" <ycao@meta.com>
Date: Thu, 19 Dec 2024 10:34:32 -0800
Subject: [PATCH] executorch/backends/apple/mps/test (#7379)

Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/7379

Reviewed By: avikchaudhuri, ydwu4

Differential Revision: D67381379
---
 backends/apple/mps/test/test_mps_utils.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/backends/apple/mps/test/test_mps_utils.py b/backends/apple/mps/test/test_mps_utils.py
index 39ce5df5115..43ae9aa0f0d 100644
--- a/backends/apple/mps/test/test_mps_utils.py
+++ b/backends/apple/mps/test/test_mps_utils.py
@@ -247,10 +247,7 @@ def lower_module_and_test_output(
             )
 
             executorch_program = to_edge(
-                export(
-                    delegated_program,
-                    sample_inputs,
-                ),
+                export(delegated_program, sample_inputs, strict=True),
                 compile_config=exir.EdgeCompileConfig(
                     _check_ir_validity=False,
                     _skip_dim_order=True,  # TODO(T182928844): Delegate dim order op to backend.