Skip to content

[Fix] docker: Pin sgl-nvidia-mega to 3333e4c so the megakernel starts on main - #46030

Merged
Oasis-Git merged 1 commit into
sgl-project:mainfrom
weireweire:fix/docker-bump-sgl-nvidia-mega
Oct 24, 2026
Merged

Oasis-Git merged 1 commit into
sgl-project:mainfrom
weireweire:fix/docker-bump-sgl-nvidia-mega

Conversation

@weireweire

@weireweire weireweire commented Oct 23, 2026 •

Copy link
Copy Markdown
Contributor

Motivation

docker/Dockerfile and docker/Dockerfile.cu134 pin sgl-nvidia-mega d47deb33. That commit predates two SGLang changes it depends on: the TopKConfig shared-expert field rename (#44912, #44913) and the kernel regrouping (#45112). As a result, it fails while building its MoE slots on current main, and every image built from main ships a megakernel that cannot start under --enable-nvidia-megakernel. This also blocks the DSpark megakernel path from #45450 in public images.

#45595 made the same bump and was closed in favor of #45389. #45389 is a larger refactor and currently changes only docker/rocm.Dockerfile. This PR is the standalone pin bump for the two CUDA Dockerfiles.

Modifications

  • Pin SGL_NVIDIA_MEGA_COMMIT to 3333e4cfac0f2b15603565d702b4c67538f1628f in docker/Dockerfile and docker/Dockerfile.cu134.
  • 3333e4cf is the plugin main commit that follows both SGLang changes (sgl-nvidia-mega Rename image_url to image_file #15, Fix for T4 GPUs #16) and includes the Engram prefetch fix (fix radix cache match #7). It keeps the nvidia_megakernel entry point, so the existing install check is unchanged.

Validation

DeepSeek-V4.1-Flash on 4x GB300, TP4/EP1, DSpark block size 5, --enable-nvidia-megakernel, plugin at this commit:

  • Every rank reports NVIDIA megakernel: active for the target and the draft.
  • GSM8K, 50 questions: 49/50.
  • AgentX trace replay at concurrency 1: 0 request errors.

pre-commit run --all-files passes.


CI States

Latest PR Test (Base): ✅ Run #39934258095
Latest PR Test (Extra): ❌ Run #39934258045
Latest PR Test (AMD ROCm 10): ➖ No AMD PR run found for this commit.

… on main

Root cause: both CUDA Dockerfiles pin sgl-nvidia-mega d47deb33. Since
sgl-project#44912/sgl-project#44913 (TopKConfig) and sgl-project#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 sgl-project#15, sgl-project#16)
and includes the Engram prefetch fix (sgl-project#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.
@Oasis-Git
Oasis-Git merged commit 49a2fbf into sgl-project:main Oct 24, 2026
109 of 113 checks passed
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