Skip to content

refactor(omni): multi-stage Docker build with decoupled builder/runtime#549

Draft
KristianZeng wants to merge 1 commit into
intel:mainfrom
KristianZeng:refactor/omni-multistage-build
Draft

refactor(omni): multi-stage Docker build with decoupled builder/runtime#549
KristianZeng wants to merge 1 commit into
intel:mainfrom
KristianZeng:refactor/omni-multistage-build

Conversation

@KristianZeng

Copy link
Copy Markdown
Contributor
  • Split BASE_IMAGE into BUILDER_IMAGE (omix:0.1.0-devel) and RUNTIME_IMAGE (pytorch:xpu-2.11.0)
  • Use Python venv (/opt/buildenv, /opt/venv) instead of system Python
  • Upgrade PyTorch to 2.11.0+xpu, unpin torchvision/torchaudio
  • Parameterize DPCPP_SYCL_TARGET and SGL_KERNEL_XPU_COMMIT
  • Pre-build sgl-kernel-xpu and omni_xpu_kernel as wheels
  • Slim runtime stage: no oneAPI compiler, only libdnnl.so.3.9
  • Add APT cache mounts for faster rebuilds
  • Fix omni_xpu_kernel setup.py: deduplicate oneDNN include paths from setvars.sh

…me images

- Split BASE_IMAGE into BUILDER_IMAGE (omix:0.1.0-devel) and RUNTIME_IMAGE (pytorch:xpu-2.11.0)
- Use Python venv (/opt/buildenv, /opt/venv) instead of system Python
- Upgrade PyTorch to 2.11.0+xpu, unpin torchvision/torchaudio
- Parameterize DPCPP_SYCL_TARGET and SGL_KERNEL_XPU_COMMIT
- Pre-build sgl-kernel-xpu and omni_xpu_kernel as wheels
- Slim runtime stage: no oneAPI compiler, only libdnnl.so.3.9
- Add APT cache mounts for faster rebuilds
- Fix omni_xpu_kernel setup.py: deduplicate oneDNN include paths from setvars.sh
@KristianZeng KristianZeng changed the title refactor(omni): multi-stage Docker build with decoupled builder/runti… refactor(omni): multi-stage Docker build with decoupled builder/runtime Jul 17, 2026
Comment thread omni/docker/Dockerfile
pip install -e . && \
rm -f /tmp/yunchang_for_multi_arc.patch /tmp/xdit_for_multi_arc.patch

# ComfyUI core (upstream, no patches)

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Restore comments.

@KristianZeng
KristianZeng marked this pull request as draft July 17, 2026 06:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants