Conv Combos with Simulator

Summary: Conv Combos that fail in simulator

Differential Revision: D66217464

Author: Eashan Garg

backends/arm/test/ops/test_conv_combos.py

@@ -253,6 +253,32 @@ def _test_conv_combo_ethos_BI_pipeline(
             .to_executorch()
         )
 
+    def _test_conv_combo_ethos_MI_pipeline(
+        self,
+        module: torch.nn.Module,
+        compile_spec: CompileSpec,
+        test_data: Tuple[torch.Tensor],
+        atol: float = 1e-3,
+        rtol: float = 1e-3,
+    ):
+        (
+            ArmTester(
+                module,
+                example_inputs=test_data,
+                compile_spec=compile_spec,
+            )
+            .quantize()
+            .export()
+            .to_edge()
+            .partition()
+            .check_count({"torch.ops.higher_order.executorch_call_delegate": 1})
+            .check_not(list(module.edge_op_list))
+            .to_executorch()
+            .run_method_and_compare_outputs(
+                inputs=test_data, atol=atol, rtol=rtol, qtol=1
+            )
+        )
+
     ####################
     ## Conv + meandim ##
     ####################
@@ -272,6 +298,14 @@ def test_conv_meandim_u55_BI(self):
             model.get_inputs(),
         )
 
+    def test_conv_meandim_u55_MI(self):
+        model = ComboConv2dMeandim()
+        self._test_conv_combo_ethos_MI_pipeline(
+            model,
+            common.get_u55_compile_spec(permute_memory_to_nhwc=True),
+            model.get_inputs(),
+        )
+
     def test_conv_meandim_u85_BI(self):
         model = ComboConv2dMeandim()
         self._test_conv_combo_ethos_BI_pipeline(
@@ -297,6 +331,12 @@ def test_conv_batchnorm_relu6_u55_BI(self):
             model, common.get_u55_compile_spec(), model.get_inputs()
         )
 
+    def test_conv_batchnorm_relu6_u55_MI(self):
+        model = ComboConvBatchnormRelu6()
+        self._test_conv_combo_ethos_MI_pipeline(
+            model, common.get_u55_compile_spec(), model.get_inputs()
+        )
+
     def test_conv_batchnorm_relu_u85_BI(self):
         model = ComboConvBatchnormRelu6()
         self._test_conv_combo_ethos_BI_pipeline(
@@ -328,6 +368,14 @@ def test_conv_relu6_u55_BI(self, test_data: torch.Tensor):
             model, common.get_u55_compile_spec(permute_memory_to_nhwc=True), test_data
         )
 
+    @parameterized.expand(ComboConvRelu6.test_data)
+    def test_conv_relu6_u55_MI(self, test_data: torch.Tensor):
+        model = ComboConvRelu6()
+        test_data = (test_data,)
+        self._test_conv_combo_ethos_MI_pipeline(
+            model, common.get_u55_compile_spec(permute_memory_to_nhwc=True), test_data
+        )
+
     @parameterized.expand(ComboConvRelu6.test_data)
     def test_conv_relu6_u85_BI(self, test_data: torch.Tensor):
         model = ComboConvRelu6()
@@ -357,6 +405,14 @@ def test_block_bottleneck_residual_u55_BI(self):
             model.get_inputs(),
         )
 
+    def test_block_bottleneck_residual_u55_MI(self):
+        model = ComboBlockBottleneckResidual()
+        self._test_conv_combo_ethos_MI_pipeline(
+            model,
+            common.get_u55_compile_spec(permute_memory_to_nhwc=True),
+            model.get_inputs(),
+        )
+
     def test_block_bottleneck_residual_u85_BI(self):
         model = ComboBlockBottleneckResidual()
         self._test_conv_combo_ethos_BI_pipeline(
@@ -390,6 +446,16 @@ def test_conv_avgpool2d_u55_BI(self, test_data: torch.Tensor):
             test_data,
         )
 
+    @parameterized.expand(ComboConvAvgPool2d.test_data)
+    def test_conv_avgpool2d_u55_MI(self, test_data: torch.Tensor):
+        model = ComboConvAvgPool2d()
+        test_data = (test_data,)
+        self._test_conv_combo_ethos_MI_pipeline(
+            model,
+            common.get_u55_compile_spec(),
+            test_data,
+        )
+
     @parameterized.expand(ComboConvAvgPool2d.test_data)
     def test_conv_avgpool2d_u85_BI(self, test_data: torch.Tensor):
         model = ComboConvAvgPool2d()