Also update cron.sh

huydhn · huydhn · commit ab228ecda104 · 2025-03-05T02:48:44.000-08:00
diff --git a/vllm-benchmarks/cron.sh b/vllm-benchmarks/cron.sh
@@ -66,7 +66,7 @@ run_benchmarks() {
 }
 
 if command -v nvidia-smi; then
-  declare -g GPU_DEVICE=$(nvidia-smi --query-gpu=name --format=csv,noheader | awk '{print $2}')
+  declare -g GPU_DEVICE=$(nvidia-smi -i 0 --query-gpu=name --format=csv,noheader | awk '{print $2}')
 elif command -v amd-smi; then
   declare -g GPU_DEVICE=$(amd-smi static -g 0 -a | grep 'MARKET_NAME' | awk '{print $2}')
 fi

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,7 @@ run_benchmarks() {`
`66`	`66`	`}`
`67`	`67`
`68`	`68`	`if command -v nvidia-smi; then`
`69`		`- declare -g GPU_DEVICE=$(nvidia-smi --query-gpu=name --format=csv,noheader \| awk '{print $2}')`
	`69`	`+ declare -g GPU_DEVICE=$(nvidia-smi -i 0 --query-gpu=name --format=csv,noheader \| awk '{print $2}')`
`70`	`70`	`elif command -v amd-smi; then`
`71`	`71`	`declare -g GPU_DEVICE=$(amd-smi static -g 0 -a \| grep 'MARKET_NAME' \| awk '{print $2}')`
`72`	`72`	`fi`