diff --git a/.github/workflows/trunk.yml b/.github/workflows/trunk.yml index 8add54af49c..ff50242f975 100644 --- a/.github/workflows/trunk.yml +++ b/.github/workflows/trunk.yml @@ -290,6 +290,7 @@ jobs: - test_arm_baremetal: test_models_ethos-u85 - test_arm_baremetal: test_smaller_stories_llama - test_arm_baremetal: test_memory_allocation + - test_arm_baremetal: test_model_smollm2-135M fail-fast: false with: runner: linux.2xlarge.memory diff --git a/backends/arm/test/test_arm_baremetal.sh b/backends/arm/test/test_arm_baremetal.sh index b8e8aee4e3a..4bc4fe0f06d 100755 --- a/backends/arm/test/test_arm_baremetal.sh +++ b/backends/arm/test/test_arm_baremetal.sh @@ -340,6 +340,19 @@ test_full_vkml() { # All End to End model tests echo "${TEST_SUITE_NAME}: PASS" } +test_model_smollm2-135M() { + echo "${TEST_SUITE_NAME}: Test SmolLM2-135M on Ethos-U85" + + # Build common libs once + python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --build_libs + + python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --target=ethos-u85-128 --model=smollm2 --extra_flags="-DEXECUTORCH_SELECT_OPS_LIST=dim_order_ops::_to_dim_order_copy.out" + + echo "${TEST_SUITE_NAME}: PASS" + + +} + test_smaller_stories_llama() { echo "${TEST_SUITE_NAME}: Test smaller_stories_llama"