make cuda process killer conditional

Signed-off-by: Ata Fatahi <[email protected]>

scripts/killall_sglang.sh

@@ -9,8 +9,11 @@ kill -9 $(ps aux | grep 'sglang.launch_server' | grep -v 'grep' | awk '{print $2
 kill -9 $(ps aux | grep 'sglang.bench' | grep -v 'grep' | awk '{print $2}') 2>/dev/null
 
 # Clean all GPU processes
-kill -9 $(nvidia-smi | sed -n '/Processes:/,$p' | grep "   [0-9]" | awk '{print $5}') 2>/dev/null
-lsof /dev/nvidia* | awk '{print $2}' | xargs kill -9 2>/dev/null
+if [ $# -gt 0 ]; then
+    kill -9 $(nvidia-smi | sed -n '/Processes:/,$p' | grep "   [0-9]" | awk '{print $5}') 2>/dev/null
+    lsof /dev/nvidia* | awk '{print $2}' | xargs kill -9 2>/dev/null
+fi
+
 
 # Show GPU status after clean up
 nvidia-smi