summaryrefslogtreecommitdiffstats
path: root/tools/module_ops/unload_all.sh
blob: 4c73a0b6dc18ef68a4e451984ecea6ccb69cf7af (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#!/bin/bash
# Unload all vGPU modules cleanly
# Usage: sudo ./unload_all.sh

set -euo pipefail

echo "=== Unloading Open-IOV vGPU Stack ==="

# Unload in reverse order of loading
modules=(
    "nvidia-vgpu-vfio"
    "nvidia-drm"
    "nvidia-modeset"
    "nvidia-uvm"
    "nvidia-peermem"
    "nvidia"
    "mdev"
)

for module in "${modules[@]}"; do
    if lsmod | grep -q "^$module "; then
        echo "Unloading $module..."
        if sudo rmmod "$module" 2>/dev/null; then
            echo "  ✓ $module unloaded"
        else
            echo "  ⚠ $module could not be unloaded (may be in use)"
            # Check refcnt
            if [[ -f /sys/module/$module/refcnt ]]; then
                echo "    refcnt: $(cat /sys/module/$module/refcnt)"
            fi
            if [[ -f /sys/module/$module/initstate ]]; then
                echo "    initstate: $(cat /sys/module/$module/initstate)"
            fi
        fi
    else
        echo "  - $module not loaded"
    fi
done

echo ""
echo "=== Unload Complete ==="
lsmod | grep -E 'nvidia|mdev' || echo "No NVIDIA/mdev modules loaded"