From cf986a34a9e461ead935c712d05918a8deff0b7a Mon Sep 17 00:00:00 2001
From: John Doe <johndoe@example.com>
Date: Wed, 4 Mar 2026 00:07:13 -0500
Subject: [PATCH] feat: add function to ensure /dev/kfd passthrough for
 LocalAGI in LXC containers

---
 ct/localagi.sh | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/ct/localagi.sh b/ct/localagi.sh
index 7d89a63ed..780ff05bd 100644
--- a/ct/localagi.sh
+++ b/ct/localagi.sh
@@ -128,6 +128,39 @@ EOF
   msg_ok "Installed ROCm runtime packages"
 }
 
+ensure_localagi_kfd_passthrough() {
+  if [[ "${var_gpu:-no}" != "yes" ]]; then
+    return 0
+  fi
+
+  if [[ ! -e /dev/kfd ]]; then
+    return 0
+  fi
+
+  local lxc_config="/etc/pve/lxc/${CTID}.conf"
+  if [[ ! -f "${lxc_config}" ]]; then
+    return 0
+  fi
+
+  if grep -qE '^dev[0-9]+: /dev/kfd(,|$)' "${lxc_config}"; then
+    return 0
+  fi
+
+  local dev_index=0
+  while grep -q "^dev${dev_index}:" "${lxc_config}"; do
+    dev_index=$((dev_index + 1))
+  done
+
+  echo "dev${dev_index}: /dev/kfd,gid=44" >>"${lxc_config}"
+  msg_ok "Added LocalAGI /dev/kfd passthrough"
+
+  if pct status "${CTID}" 2>/dev/null | grep -q running; then
+    msg_info "Restarting container to apply /dev/kfd passthrough"
+    pct reboot "${CTID}" >/dev/null 2>&1 || true
+    msg_ok "Container restart requested"
+  fi
+}
+
 function update_script() {
   # Standard update prechecks and environment summary.
   header_info
@@ -212,6 +245,7 @@ function update_script() {
 
 start
 build_container
+ensure_localagi_kfd_passthrough
 description
 
 msg_ok "Completed successfully!\n"