args

Author: kirklandsign

extension/benchmark/android/benchmark/android-llm-device-farm-test-spec.yml.j2

@@ -116,11 +116,13 @@ phases:
         if [ -n "$BIN_FOUND" ]; then
           adb -s $DEVICEFARM_DEVICE_UDID shell am start -W -n org.pytorch.minibench/.BenchmarkActivity \
             --es "model_dir" "/data/local/tmp/minibench" \
-            --es "tokenizer_path" "/data/local/tmp/minibench/tokenizer.bin"
+            --es "tokenizer_path" "/data/local/tmp/minibench/tokenizer.bin" \
+            --ei "num_iter" 5 --ei "num_warm_up_iter" 2
         elif [ -n "$MODEL_FOUND" ]; then
           adb -s $DEVICEFARM_DEVICE_UDID shell am start -W -n org.pytorch.minibench/.BenchmarkActivity \
             --es "model_dir" "/data/local/tmp/minibench" \
-            --es "tokenizer_path" "/data/local/tmp/minibench/tokenizer.model"
+            --es "tokenizer_path" "/data/local/tmp/minibench/tokenizer.model" \
+            --ei "num_iter" 5 --ei "num_warm_up_iter" 2
         else
           adb -s $DEVICEFARM_DEVICE_UDID shell am start -W -n org.pytorch.minibench/.BenchmarkActivity \
             --es "model_dir" "/data/local/tmp/minibench"