Arm backend: Add smaller_stories_llama to test_arm_baremetal

Run it in trunk job

Signed-off-by: Erik Lundell <[email protected]>
Change-Id: Ia1ba80330c7ff457ff06f1975ecfd01262f58457

Author: Erik-Lundell

.github/workflows/trunk.yml

@@ -234,6 +234,7 @@ jobs:
           - test_arm_baremetal: test_models_tosa
           - test_arm_baremetal: test_models_ethos-u55
           - test_arm_baremetal: test_models_ethos-u85
+          - test_arm_baremetal: test_smaller_stories_llama
       fail-fast: false
     with:
       runner: linux.2xlarge.memory

backends/arm/test/test_arm_baremetal.sh

@@ -253,6 +253,31 @@ test_full_ethosu_fvp() { # All End to End model tests
     echo "${TEST_SUITE_NAME}: PASS"
     }
 
+test_smaller_stories_llama() {
+    echo "${TEST_SUITE_NAME}: Test smaller_stories_llama"
+
+    backends/arm/scripts/build_executorch.sh
+
+    mkdir -p stories110M
+    pushd stories110M
+    wget -N https://huggingface.co/karpathy/tinyllamas/resolve/main/stories110M.pt
+    echo '{"dim": 768, "multiple_of": 32, "n_heads": 12, "n_layers": 12, "norm_eps": 1e-05, "vocab_size": 32000}' > params.json
+    popd
+
+    # Get path to source directory
+    pytest \
+    -c /dev/null \
+    --verbose \
+    --color=yes \
+    --numprocesses=auto \
+    --log-level=DEBUG \
+    --junit-xml=stories110M/test-reports/unittest.xml \
+    -s \
+    backends/arm/test/models/test_llama.py \
+    --llama_inputs stories110M/stories110M.pt stories110M/params.json stories110m
+
+    echo "${TEST_SUITE_NAME}: PASS"
+    }
 
 
 ${TEST_SUITE}