Skip to content

Commit b472f1b

Browse files
Arm backend: Add SmolLM2-135M to CI model testing
Change-Id: Ib29b8c52b9f3ad16d861fcf0b49d79e197053694 Signed-off-by: Sebastian Larsson <[email protected]>
1 parent ce6e2cf commit b472f1b

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

.github/workflows/trunk.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,7 @@ jobs:
289289
- test_arm_baremetal: test_models_ethos-u55
290290
- test_arm_baremetal: test_models_ethos-u85
291291
- test_arm_baremetal: test_smaller_stories_llama
292+
- test_arm_baremetal: test_model_smollm2-135M
292293
fail-fast: false
293294
with:
294295
runner: linux.2xlarge.memory

backends/arm/test/test_arm_baremetal.sh

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -340,6 +340,19 @@ test_full_vkml() { # All End to End model tests
340340
echo "${TEST_SUITE_NAME}: PASS"
341341
}
342342

343+
test_model_smollm2-135M() {
344+
echo "${TEST_SUITE_NAME}: Test SmolLM2-135M on Ethos-U85"
345+
346+
# Build common libs once
347+
python3 backends/arm/test/test_model.py --test_output=arm_test/test_model --build_libs
348+
349+
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"
350+
351+
echo "${TEST_SUITE_NAME}: PASS"
352+
353+
354+
}
355+
343356
test_smaller_stories_llama() {
344357
echo "${TEST_SUITE_NAME}: Test smaller_stories_llama"
345358

0 commit comments

Comments
 (0)