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:
1 parent
aa38c556b2
commit
354ddc473f
3 files changed
+50
-14
No files matched your search
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user