diff --git a/backends/arm/test/ops/test_cat.py b/backends/arm/test/ops/test_cat.py index a40ae43b673..63dbc327d1e 100644 --- a/backends/arm/test/ops/test_cat.py +++ b/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) diff --git a/backends/arm/test/ops/test_expand.py b/backends/arm/test/ops/test_expand.py index 66c081a544c..9058628c4bb 100644 --- a/backends/arm/test/ops/test_expand.py +++ b/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)) diff --git a/backends/arm/test/ops/test_repeat.py b/backends/arm/test/ops/test_repeat.py index a6fad033456..50d1b224ad5 100644 --- a/backends/arm/test/ops/test_repeat.py +++ b/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)) diff --git a/backends/arm/test/ops/test_slice.py b/backends/arm/test/ops/test_slice.py index 14874df156e..b8f15f650cf 100644 --- a/backends/arm/test/ops/test_slice.py +++ b/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,)) diff --git a/backends/arm/test/ops/test_split.py b/backends/arm/test/ops/test_split.py index bc998179c0c..23153253adb 100644 --- a/backends/arm/test/ops/test_split.py +++ b/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) diff --git a/examples/arm/aot_arm_compiler.py b/examples/arm/aot_arm_compiler.py index 4d77e819089..220759bd2d2 100644 --- a/examples/arm/aot_arm_compiler.py +++ b/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() ) diff --git a/examples/arm/executor_runner/arm_executor_runner.cpp b/examples/arm/executor_runner/arm_executor_runner.cpp index b6ac43bcf21..aa126237849 100644 --- a/examples/arm/executor_runner/arm_executor_runner.cpp +++ b/examples/arm/executor_runner/arm_executor_runner.cpp @@ -105,10 +105,12 @@ Result 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(allocator.allocate(num_inputs * sizeof(void*))); diff --git a/examples/arm/setup.sh b/examples/arm/setup.sh index 8c39a3a8668..a59fd1eaace 100755 --- a/examples/arm/setup.sh +++ b/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"