ComfyUI: GPU (gfx1151 ROCm) build via container

Replace the CPU stock image with a container built from
Containerfile.comfyui on AMD's gfx1151-tagged vLLM base — the only torch
build verified to report torch.cuda.is_available()=True on the Strix Halo
APU (Radeon 8060S, capability (11,5)).

- Containerfile bakes the ComfyUI code in and installs its deps, but skips
  the torch family (the base image already ships the gfx1151 ROCm
  torch/torchvision/torchaudio; an unpinned 'torch' in requirements.txt would
  clobber the gfx1151 kernels).
- compose comfyui service: build + image comfyui-rocm:gfx1151, passes
  /dev/kfd + /dev/dri, adds video+render groups, drops --cpu. Models/data
  persist under /opt/comfyui.
- Un-skip the four compose-wiring acceptance tests (the gate for this work).
This commit is contained in:
Frank committed 2026-08-19 15:20:46 -07:00
1 parent aa38c556b2
commit 354ddc473f
3 files changed
+50 -14

No files matched your search

+5 -9
View File
@@ -7,16 +7,15 @@
# - NOT pass --cpu
# - add the video and render groups
#
# STATUS (2026-08-18): the GPU migration is NOT yet applied. The compose still
# runs ComfyUI with --cpu on the stock image (the gfx1151 torch image is not
# built yet). The four compose-wiring tests below are skipped until the ROCm
# build lands. The Containerfile test stays active: the file exists and is ready.
# See docs/Research/ComfyUI ROCm and Vulkan.md. Re-enable the skips then.
# STATUS (2026-08-19): the GPU migration is applied. The compose builds
# comfyui-rocm:gfx1151 from Containerfile.comfyui (AMD's gfx1151 torch base),
# passes /dev/kfd + /dev/dri, adds the video+render groups, and drops --cpu.
# The four compose-wiring tests below are active (un-skipped). See
# docs/Research/ComfyUI ROCm and Vulkan.md.
load helpers
@test "comfyui builds from the gfx1151 ROCm Containerfile" {
skip "GPU ComfyUI migration pending (CPU path in use)"
local compose="$REPO_ROOT/dankclaw/compose.yaml"
python3 - "$compose" <<'PYEOF'
import sys, yaml
@@ -39,7 +38,6 @@ PYEOF
}
@test "comfyui passes /dev/kfd and /dev/dri to the container" {
skip "GPU ComfyUI migration pending (CPU path in use)"
python3 - "$REPO_ROOT/dankclaw/compose.yaml" <<'PYEOF'
import sys, yaml
with open(sys.argv[1]) as fh:
@@ -53,7 +51,6 @@ PYEOF
}
@test "comfyui does not run with --cpu" {
skip "GPU ComfyUI migration pending (CPU path in use)"
python3 - "$REPO_ROOT/dankclaw/compose.yaml" <<'PYEOF'
import sys, yaml
with open(sys.argv[1]) as fh:
@@ -67,7 +64,6 @@ PYEOF
}
@test "comfyui adds the video and render groups" {
skip "GPU ComfyUI migration pending (CPU path in use)"
python3 - "$REPO_ROOT/dankclaw/compose.yaml" <<'PYEOF'
import sys, yaml
with open(sys.argv[1]) as fh: