Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .github/workflows/trunk.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
13 changes: 13 additions & 0 deletions backends/arm/test/test_arm_baremetal.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

curious why do we need dim_order_copy (i.e. why do we have a graph break)..

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Did you guys run it on a real hardware? what's the TPS looks like?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

No this was not run on real hardware.


echo "${TEST_SUITE_NAME}: PASS"


}

test_smaller_stories_llama() {
echo "${TEST_SUITE_NAME}: Test smaller_stories_llama"

Expand Down
Loading