diff --git a/python/sglang/utils.py b/python/sglang/utils.py
index c1bf62ef983..5689b097d23 100644
--- a/python/sglang/utils.py
+++ b/python/sglang/utils.py
@@ -1,4 +1,4 @@
-"""Common utilities."""
+"""Common utilities"""
 
 import base64
 import gc
diff --git a/scripts/killall_sglang.sh b/scripts/killall_sglang.sh
index fcad493c59c..cb187f46a0f 100755
--- a/scripts/killall_sglang.sh
+++ b/scripts/killall_sglang.sh
@@ -1,5 +1,14 @@
-# Kill all SGLang processes and free the GPU memory.
+#!/bin/bash
 
-kill -9 $(ps aux | grep 'multiprocessing.spawn' | grep -v 'grep' | awk '{print $2}')
-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}')
+# Show current GPU status
+nvidia-smi
+
+# Clean SGLang processes
+kill -9 $(ps aux | grep 'multiprocessing.spawn' | 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
+
+# Clean all GPU processes if any argument is provided
+if [ $# -gt 0 ]; then
+    kill -9 $(nvidia-smi | sed -n '/Processes:/,$p' | grep "   [0-9]" | awk '{print $5}') 2>/dev/null
+fi