#!/bin/sh
# 9001-amd-rocm.hook.chroot — install AMD ROCm SMI tool for Instinct GPU monitoring.
# Runs inside the live-build chroot. Adds AMD's apt repository and installs
# rocm-smi-lib which provides the `rocm-smi` CLI (analogous to nvidia-smi).

set -e

ROCM_VERSION="6.4"
ROCM_KEYRING="/etc/apt/keyrings/rocm.gpg"
ROCM_LIST="/etc/apt/sources.list.d/rocm.list"

echo "=== AMD ROCm ${ROCM_VERSION}: adding repository ==="

mkdir -p /etc/apt/keyrings

# Download and import AMD GPG key
if ! wget -qO- "https://repo.radeon.com/rocm/rocm.gpg.key" \
        | gpg --dearmor > "${ROCM_KEYRING}"; then
    echo "WARN: failed to fetch AMD ROCm GPG key — skipping ROCm install"
    exit 0
fi

cat > "${ROCM_LIST}" <<EOF
deb [arch=amd64 signed-by=${ROCM_KEYRING}] https://repo.radeon.com/rocm/apt/${ROCM_VERSION} bookworm main
EOF

apt-get update -qq

# rocm-smi-lib provides the rocm-smi CLI tool for GPU monitoring
if apt-get install -y --no-install-recommends rocm-smi-lib 2>/dev/null; then
    echo "=== AMD ROCm: rocm-smi installed ==="
    rocm-smi --version 2>/dev/null || true
else
    echo "WARN: rocm-smi-lib install failed — GPU monitoring unavailable"
fi

# Clean up apt lists to keep ISO size down
rm -f "${ROCM_LIST}"
apt-get clean