Try with any pte

pytorch · Sep 11, 2024 · 64f1476 · 64f1476
1 parent 5214824
commit 64f1476
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/extension/android/benchmark/android-llm-device-farm-test-spec.yml b/extension/android/benchmark/android-llm-device-farm-test-spec.yml
@@ -80,6 +80,12 @@ phases:
         --es "model_dir" "/data/local/tmp/llama" \
         --es "tokenizer_path" "/data/local/tmp/llama/tokenizer.bin"
 
+      - echo "Run generic benchmark"
+      - |
+        adb -s $DEVICEFARM_DEVICE_UDID shell am start -W -n org.pytorch.minibench/.BenchmarkActivity \
+        --es "model_dir" "/data/local/tmp/llama"
+
+
   post_test:
     commands:
       - echo "Gather LLM benchmark results"