From d8b2a372284b24480c771cb71e8f02fd03f8e23b Mon Sep 17 00:00:00 2001
From: "CanbiZ (MickLesk)" <47820557+MickLesk@users.noreply.github.com>
Date: Mon, 9 Mar 2026 10:18:08 +0100
Subject: [PATCH] fix(build): auto-resize disk +8GB when AMD GPU detected for
 ROCm

---
 misc/build.func | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/misc/build.func b/misc/build.func
index 706e5b57c..b33e96818 100644
--- a/misc/build.func
+++ b/misc/build.func
@@ -3922,6 +3922,17 @@ EOF
   configure_gpu_passthrough
   configure_additional_devices
 
+  # Increase disk size for AMD ROCm compute stack (~8GB extra needed)
+  if [[ "${GPU_TYPE:-}" == "AMD" ]]; then
+    local rocm_extra=8
+    local new_disk_size=$((PCT_DISK_SIZE + rocm_extra))
+    if pct resize "$CTID" rootfs "${new_disk_size}G" >/dev/null 2>&1; then
+      msg_ok "Disk resized ${PCT_DISK_SIZE}GB → ${new_disk_size}GB for ROCm"
+    else
+      msg_warn "Failed to resize disk for ROCm — installation may fail if space is insufficient"
+    fi
+  fi
+
   # ============================================================================
   # START CONTAINER AND INSTALL USERLAND
   # ============================================================================