From e54d4ced6d43c4f2eed6980dbd93892a1c72f2c6 Mon Sep 17 00:00:00 2001 From: Zingo Andersen Date: Mon, 2 Dec 2024 09:46:29 +0100 Subject: [PATCH] Arm backend : Back Ethos-U Vela compiler to working version Latest vela version 4.1.0 unfortunately generates output size/shapes not matching the expected output. Moving the Vela version back to a working version until this is investigated/fixed. Signed-off-by: Zingo Andersen Change-Id: I8ab4766b9a0de22779cdf8c9932d78eb6e6e87e5 --- backends/arm/test/ops/test_depthwise_conv.py | 1 - examples/arm/aot_arm_compiler.py | 4 ++-- examples/arm/setup.sh | 2 +- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/backends/arm/test/ops/test_depthwise_conv.py b/backends/arm/test/ops/test_depthwise_conv.py index 083e9aaf68..dafad5803f 100644 --- a/backends/arm/test/ops/test_depthwise_conv.py +++ b/backends/arm/test/ops/test_depthwise_conv.py @@ -260,7 +260,6 @@ def test_dw_conv_tosa_BI(self, test_name: str, model: torch.nn.Module): ) # Works @parameterized.expand(testsuite_conv2d, skip_on_empty=True) - @unittest.expectedFailure def test_dw_conv2d_u55_BI( self, test_name: str, model: torch.nn.Module, set_quantize_io: bool = False ): diff --git a/examples/arm/aot_arm_compiler.py b/examples/arm/aot_arm_compiler.py index 6d899c2146..484d24ff5b 100644 --- a/examples/arm/aot_arm_compiler.py +++ b/examples/arm/aot_arm_compiler.py @@ -273,7 +273,7 @@ def get_compile_spec( target, system_config="Ethos_U55_High_End_Embedded", memory_mode="Shared_Sram", - extra_flags="--debug-force-regor --output-format=raw --verbose-operators --verbose-cycle-estimate", + extra_flags="--debug-force-regor --output-format=raw", ) .set_permute_memory_format(True) .set_quantize_io(True) @@ -286,7 +286,7 @@ def get_compile_spec( target, system_config="Ethos_U85_SYS_DRAM_Mid", memory_mode="Shared_Sram", - extra_flags="--output-format=raw --verbose-operators --verbose-cycle-estimate", + extra_flags="--output-format=raw", ) .set_permute_memory_format(True) .set_quantize_io(True) diff --git a/examples/arm/setup.sh b/examples/arm/setup.sh index 6f619ef058..ebcc6d66f2 100755 --- a/examples/arm/setup.sh +++ b/examples/arm/setup.sh @@ -92,7 +92,7 @@ tosa_reference_model_rev="f9ea4ab7da19318fe36b1c34d68a3e40fd6e56c5" # vela vela_repo_url="https://review.mlplatform.org/ml/ethos-u/ethos-u-vela" -vela_rev="a08fc18780827b5fefc814dd0162ee6317ce0ae7" +vela_rev="49d0ae16bafd1db0fcb6790f12b7f249acf1974d" ######## ### Mandatory user args