From 032b887ced20c96f0da50184f8246647efef59bd Mon Sep 17 00:00:00 2001
From: Clemens Akens <clebert@me.com>
Date: Wed, 25 Oct 2023 14:04:10 +0200
Subject: [PATCH] Update benchmark script

---
 benchmark.sh | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/benchmark.sh b/benchmark.sh
index a0aa1c9..d4ec573 100755
--- a/benchmark.sh
+++ b/benchmark.sh
@@ -6,16 +6,17 @@ set -u           # Triggers an error when an unset variable is called
 set -o noclobber # Prevents from overwriting existing files
 
 if [ "$#" -ne 2 ]; then
-    echo "Usage: ./benchmark.sh <model> <runs>"
+    echo "Usage: ./benchmark.sh <model> <max_worker_count> <runs>"
     exit 1
 fi
 
 model=$1
-runs=$2
+max_worker_count=$2
+runs=$3
 
 zig build -Doptimize=ReleaseFast
 
-for ((worker_count=0; worker_count<11; worker_count++))
+for ((worker_count=0; worker_count<=max_worker_count; worker_count++))
 do
   echo -n "Running $model with $worker_count workers:"