Arm run.sh fix with working unit tests (#6056)

* Arm run.sh fixes (#6043)

Fixes for vela compiler version, codegen flags and non-semihosted error behaviour

* enable passing tests and remove broken split test.

Signed-off-by: Rob Elliott <[email protected]>

---------

Signed-off-by: Rob Elliott <[email protected]>

Author: robell

backends/arm/test/ops/test_cat.py

@@ -127,7 +127,6 @@ def test_cat_tosa_BI(self, operands: tuple[torch.Tensor, ...], dim: int):
 
     # TODO: Remove @unittest.expectedFailure when this issue is fixed in Regor
     @parameterized.expand(Cat.test_parameters)
-    @unittest.expectedFailure
     def test_cat_u55_BI(self, operands: tuple[torch.Tensor, ...], dim: int):
         test_data = (operands, dim)
         self._test_cat_u55_BI_pipeline(self.Cat(), test_data)

backends/arm/test/ops/test_expand.py

@@ -104,6 +104,5 @@ def test_expand_tosa_BI(self, test_input, multiples):
 
     # Expected failure since tosa.TILE is unsupported by Vela.
     @parameterized.expand(Expand.test_parameters)
-    @unittest.expectedFailure
     def test_expand_u55_BI(self, test_input, multiples):
         self._test_expand_tosa_u55_pipeline(self.Expand(), (test_input, multiples))

backends/arm/test/ops/test_repeat.py

@@ -105,6 +105,5 @@ def test_repeat_tosa_BI(self, test_input, multiples):
 
     # Expected failure since tosa.TILE is unsupported by Vela.
     @parameterized.expand(Repeat.test_parameters)
-    @unittest.expectedFailure
     def test_repeat_u55_BI(self, test_input, multiples):
         self._test_repeat_tosa_u55_pipeline(self.Repeat(), (test_input, multiples))

backends/arm/test/ops/test_slice.py

@@ -111,6 +111,5 @@ def test_slice_nhwc_tosa_BI(self, test_tensor: torch.Tensor):
     # Fails during Vela compilation when trying to use a Tuple as a Named tuple,
     # Could be Vela Issue, wait until Regor.
     @parameterized.expand(Slice.test_tensors)
-    @unittest.expectedFailure
     def test_slice_u55_BI(self, test_tensor: torch.Tensor):
         self._test_slice_u55_BI_pipeline(self.Slice(), (test_tensor,))

backends/arm/test/ops/test_split.py

@@ -130,10 +130,3 @@ def test_split_n_out_tosa_MI(self, test_data: test_data_t):
     @parameterized.expand(Split.test_data)
     def test_split_tosa_BI(self, test_data: test_data_t):
         self._test_split_tosa_BI_pipeline(self.Split(), test_data)
-
-    # Fails during Vela compilation when trying to use a Tuple as a Named tuple,
-    # Could be Vela Issue, wait until Regor.
-    @parameterized.expand(Split.test_data)
-    @unittest.expectedFailure
-    def test_split_u55_BI(self, test_data: test_data_t):
-        self._test_split_u55_BI_pipeline(self.Split(), test_data)

examples/arm/aot_arm_compiler.py

@@ -214,7 +214,12 @@ def forward(self, x):
         edge = edge.to_backend(
             ArmPartitioner(
                 ArmCompileSpecBuilder()
-                .ethosu_compile_spec("ethos-u55-128")
+                .ethosu_compile_spec(
+                    "ethos-u55-128",
+                    system_config="Ethos_U55_High_End_Embedded",
+                    memory_mode="Shared_Sram",
+                    extra_flags="--debug-force-regor --output-format=raw",
+                )
                 .set_permute_memory_format(
                     args.model_name in MODEL_NAME_TO_MODEL.keys()
                 )

examples/arm/executor_runner/arm_executor_runner.cpp

@@ -105,10 +105,12 @@ Result<BufferCleanup> prepare_input_tensors(
   size_t num_inputs = method_meta.num_inputs();
   size_t num_allocated = 0;
 
+#ifdef SEMIHOSTING
   ET_CHECK_OR_RETURN_ERROR(
       input_buffers.size() > 0 && num_inputs == input_buffers.size(),
       InvalidArgument,
       "Wrong number of inputs allocated compared to method");
+#endif
 
   void** inputs =
       static_cast<void**>(allocator.allocate(num_inputs * sizeof(void*)));

examples/arm/setup.sh

@@ -216,7 +216,7 @@ function setup_vela() {
     if [[ ! -e ethos-u-vela ]]; then
         git clone https://review.mlplatform.org/ml/ethos-u/ethos-u-vela
         repo_dir="${root_dir}/ethos-u-vela"
-        base_rev=d362f5443f67b1e6213a9d8f124edff758efac96
+        base_rev=57ce18c89ccc6f6309333dccb24ed30dc68b571f
         patch_repo
     fi
     cd "${root_dir}/ethos-u-vela"