Skip to content

Commit 6f5ce2d

Browse files
Arm backend: Add SmolLM2-135M to CI model testing (#14722)
Signed-off-by: Sebastian Larsson <[email protected]>
1 parent b30ad01 commit 6f5ce2d

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
@@ -290,6 +290,7 @@ jobs:
290290
- test_arm_baremetal: test_models_ethos-u85
291291
- test_arm_baremetal: test_smaller_stories_llama
292292
- test_arm_baremetal: test_memory_allocation
293+
- test_arm_baremetal: test_model_smollm2-135M
293294
fail-fast: false
294295
with:
295296
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)