diff --git a/backends/arm/scripts/build_executor_runner.sh b/backends/arm/scripts/build_executor_runner.sh index 9e2f3954c53..807821d427f 100755 --- a/backends/arm/scripts/build_executor_runner.sh +++ b/backends/arm/scripts/build_executor_runner.sh @@ -103,7 +103,7 @@ then memory_mode="Shared_Sram" if [[ ${target} =~ "ethos-u85" ]] then - memory_mode="Dedicated_Sram_384KB" + memory_mode="Sram_Only" fi fi diff --git a/backends/arm/test/test_arm_baremetal.sh b/backends/arm/test/test_arm_baremetal.sh index 413e274da8f..12a37315109 100755 --- a/backends/arm/test/test_arm_baremetal.sh +++ b/backends/arm/test/test_arm_baremetal.sh @@ -210,7 +210,10 @@ test_models_ethos-u85() { # End to End model tests using model_test.py python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-512 --model=mv3 --extra_flags="-DET_ATOL=5.00 -DET_RTOL=5.00" python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-128 --model=lstm --extra_flags="-DET_ATOL=0.03 -DET_RTOL=0.03" python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-128 --model=w2l --extra_flags="-DET_ATOL=0.01 -DET_RTOL=0.01" - python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-256 --model=ic4 --extra_flags="-DET_ATOL=0.8 -DET_RTOL=0.8" --timeout=2400 + # Temporarily not test inception_v4 on Ethos-U85. To support inception_v4 properly on Ethos-U85, we need to run the model in Dedicated_Sram memory mode with + # 384KB(or another amount lower than 2MB) of SRAM passed as fast scratch area. The PR adding support for Dedicated_Sram(https://github.com/pytorch/executorch/pull/10714) + # was reverted due to a change required in an internal variant of the examples/arm/executor_runner/arm_executor_runner.cpp + # python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-256 --model=ic4 --extra_flags="-DET_ATOL=0.8 -DET_RTOL=0.8" --timeout=2400 echo "${TEST_SUITE_NAME}: PASS" } diff --git a/backends/arm/test/test_model.py b/backends/arm/test/test_model.py index 072583ef862..b0fd2f2a381 100755 --- a/backends/arm/test/test_model.py +++ b/backends/arm/test/test_model.py @@ -81,7 +81,7 @@ def get_args(): if "u55" in args.target: args.memory_mode = "Shared_Sram" elif "u85" in args.target: - args.memory_mode = "Dedicated_Sram_384KB" + args.memory_mode = "Sram_Only" else: raise RuntimeError(f"Invalid target name {args.target}") diff --git a/examples/arm/run.sh b/examples/arm/run.sh index 750c251596c..89ac5cd30a8 100755 --- a/examples/arm/run.sh +++ b/examples/arm/run.sh @@ -110,7 +110,7 @@ then memory_mode="Shared_Sram" if [[ ${target} =~ "ethos-u85" ]] then - memory_mode="Dedicated_Sram_384KB" + memory_mode="Sram_Only" fi fi