Fabian
48 lines
1.6 KiB
Bash
48 lines
1.6 KiB
Bash
#!/bin/bash
|
|
pid=$(pgrep -f "python.*main.py" | head -1)
|
|
echo "PID: $pid"
|
|
|
|
# Check ALL threads for running state
|
|
echo "=== All thread states ==="
|
|
for tid in $(ls /proc/$pid/task/ 2>/dev/null); do
|
|
stat=$(cat /proc/$pid/task/$tid/stat 2>/dev/null)
|
|
state=$(echo "$stat" | awk '{print $3}')
|
|
cpu=$(echo "$stat" | awk '{print $14+$15}') # utime+stime
|
|
wchan=$(cat /proc/$pid/task/$tid/wchan 2>/dev/null)
|
|
if [ "$state" = "R" ] || [ "$cpu" -gt 1000 ] 2>/dev/null; then
|
|
echo " ** TID $tid: state=$state cpu=$cpu wchan=$wchan **"
|
|
fi
|
|
done
|
|
|
|
echo ""
|
|
echo "=== Running threads only ==="
|
|
for tid in $(ls /proc/$pid/task/ 2>/dev/null); do
|
|
state=$(cat /proc/$pid/task/$tid/stat 2>/dev/null | awk '{print $3}')
|
|
if [ "$state" = "R" ]; then
|
|
echo " RUNNING: TID $tid"
|
|
cat /proc/$pid/task/$tid/wchan 2>/dev/null
|
|
fi
|
|
done
|
|
|
|
echo ""
|
|
echo "=== Top CPU threads (last column is cumulative CPU ticks) ==="
|
|
for tid in $(ls /proc/$pid/task/ 2>/dev/null); do
|
|
stat=$(cat /proc/$pid/task/$tid/stat 2>/dev/null)
|
|
utime=$(echo "$stat" | awk '{print $14}')
|
|
stime=$(echo "$stat" | awk '{print $15}')
|
|
total=$((utime + stime))
|
|
state=$(echo "$stat" | awk '{print $3}')
|
|
if [ "$total" -gt 100 ]; then
|
|
echo " TID $tid: state=$state total_ticks=$total"
|
|
fi
|
|
done
|
|
|
|
echo ""
|
|
echo "=== py-spy dump (venv) ==="
|
|
echo 0 | sudo tee /proc/sys/kernel/yama/ptrace_scope > /dev/null 2>&1
|
|
/home/fabian/ComfyUI/venv/bin/py-spy dump --pid $pid 2>&1 | head -80
|
|
|
|
echo ""
|
|
echo "=== VRAM check ==="
|
|
rocm-smi --showmeminfo vram 2>/dev/null | grep -E "Used|Total"
|