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.