We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a693dce commit 696f35aCopy full SHA for 696f35a
vllm-benchmarks/run.sh
@@ -107,7 +107,7 @@ export HEAD_BRANCH=main
107
export HEAD_SHA=$(git rev-parse --verify HEAD)
108
109
if command -v nvidia-smi; then
110
- declare -g GPU_DEVICE=$(nvidia-smi -i 0 --query-gpu=name --format=csv,noheader | awk '{print $2}' | head -n1)
+ declare -g GPU_DEVICE=$(nvidia-smi -i 0 --query-gpu=name --format=csv,noheader | awk '{print $2}')
111
elif command -v amd-smi; then
112
declare -g GPU_DEVICE=$(amd-smi static -g 0 -a | grep 'MARKET_NAME' | awk '{print $2}')
113
fi
0 commit comments