squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Nightly benchmarking suite] Remove pkill python from run benchmark suite (#6965)

Browse Source
This commit is contained in:
Cade Daniel 2024-07-30 16:28:05 -07:00 committed by GitHub
parent 40c27a7cbb
commit 79319cedfa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.buildkite/nightly-benchmarks/run-benchmarks-suite.sh
View File

@ -73,11 +73,6 @@ kill_gpu_processes() {
echo "All GPU processes have been killed." echo "All GPU processes have been killed."
fi fi
# Sometimes kill with pid doesn't work properly, we can also kill all process running python or python3
# since we are in container anyway
pkill -9 -f python
pkill -9 -f python3
# waiting for GPU processes to be fully killed # waiting for GPU processes to be fully killed
# loop while nvidia-smi returns any processes # loop while nvidia-smi returns any processes
while [ -n "$(nvidia-smi --query-compute-apps=pid --format=csv,noheader)" ]; do while [ -n "$(nvidia-smi --query-compute-apps=pid --format=csv,noheader)" ]; do