From 4608527f48230e13cc5d0a90e8a5af19a6f8c173 Mon Sep 17 00:00:00 2001 From: weireweire <20922698+weireweire@users.noreply.github.com> Date: Fri, 23 Oct 2026 17:43:08 +0800 Subject: [PATCH] [Fix] docker: Pin sgl-nvidia-mega to 3333e4c so the megakernel starts on main Root cause: both CUDA Dockerfiles pin sgl-nvidia-mega d47deb33. Since #44912/#44913 (TopKConfig) and #45112 (kernel regrouping), that commit fails while building its MoE slots, so images built from main ship a megakernel that cannot start. Fix: pin 3333e4cf in docker/Dockerfile and docker/Dockerfile.cu134. It is the plugin main commit that follows both changes (sgl-nvidia-mega #15, #16) and includes the Engram prefetch fix (#7). Validation: DeepSeek-V4.1-Flash on 4x GB300, TP4, megakernel enabled: every rank reports the megakernel active; GSM8K-50 49/50; an AgentX c1 trace replay runs with 0 request errors. --- docker/Dockerfile | 2 +- docker/Dockerfile.cu134 | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 8ee161b8a0e2..2975f94cf8e6 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -631,7 +631,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \ # SM103, compiled at runtime); disable with --build-arg ENABLE_SGL_NVIDIA_MEGA=0. ARG ENABLE_SGL_NVIDIA_MEGA=1 ARG SGL_NVIDIA_MEGA_REPO="https://github.com/sgl-project/sgl-nvidia-mega.git" -ARG SGL_NVIDIA_MEGA_COMMIT="d47deb3390e14741c9420ac702954e7611a14da1" +ARG SGL_NVIDIA_MEGA_COMMIT="3333e4cfac0f2b15603565d702b4c67538f1628f" RUN --mount=type=cache,target=/root/.cache/uv \ if [ "${ENABLE_SGL_NVIDIA_MEGA}" != "1" ]; then \ echo "[sgl-nvidia-mega] skip (ENABLE_SGL_NVIDIA_MEGA=${ENABLE_SGL_NVIDIA_MEGA})"; \ diff --git a/docker/Dockerfile.cu134 b/docker/Dockerfile.cu134 index 40f899436cab..a7ffa29e3ad5 100644 --- a/docker/Dockerfile.cu134 +++ b/docker/Dockerfile.cu134 @@ -885,7 +885,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \ # SM103, compiled at runtime); disable with --build-arg ENABLE_SGL_NVIDIA_MEGA=0. ARG ENABLE_SGL_NVIDIA_MEGA=1 ARG SGL_NVIDIA_MEGA_REPO="https://github.com/sgl-project/sgl-nvidia-mega.git" -ARG SGL_NVIDIA_MEGA_COMMIT="d47deb3390e14741c9420ac702954e7611a14da1" +ARG SGL_NVIDIA_MEGA_COMMIT="3333e4cfac0f2b15603565d702b4c67538f1628f" RUN --mount=type=cache,target=/root/.cache/pip \ if [ "${ENABLE_SGL_NVIDIA_MEGA}" != "1" ]; then \ echo "[sgl-nvidia-mega] skip (ENABLE_SGL_NVIDIA_MEGA=${ENABLE_SGL_NVIDIA_MEGA})"; \