diff --git a/scripts/killall_sglang.sh b/scripts/killall_sglang.sh index fdafbbb98..843ee5fd3 100755 --- a/scripts/killall_sglang.sh +++ b/scripts/killall_sglang.sh @@ -4,10 +4,7 @@ nvidia-smi # Clean SGLang processes -kill -9 $(ps aux | grep 'sglang::' | grep -v 'grep' | awk '{print $2}') 2>/dev/null -kill -9 $(ps aux | grep 'sglang.launch_server' | grep -v 'grep' | awk '{print $2}') 2>/dev/null -kill -9 $(ps aux | grep 'sglang.bench' | grep -v 'grep' | awk '{print $2}') 2>/dev/null -kill -9 $(ps aux | grep 'sglang.data_parallel' | grep -v 'grep' | awk '{print $2}') 2>/dev/null +pgrep -f 'sglang::|sglang\.launch_server|sglang\.bench|sglang\.data_parallel|sglang\.srt' | xargs -r kill -9 # Clean all GPU processes if any argument is provided if [ $# -gt 0 ]; then