Pre-built SGLang image for NVIDIA Blackwell (SM120) GPUs with luke's custom PCIe allreduce kernel and RMSNorm fusion.
24994611c ("Move pcie oneshot to it's own flags")nvidia/cuda:13.0.1-cudnn-devel-ubuntu24.04 (sha256:92a047cf4837)CUDA_VERSION=13.0.1, GRACE_BLACKWELL=1, BRANCH_TYPE=local# Built from luke's SGLang fork with Blackwell support
# https://github.com/lukealonso/sglang @ 24994611c
FROM nvidia/cuda:13.0.1-cudnn-devel-ubuntu24.04
# See full Dockerfile at:
# https://github.com/lukealonso/sglang/blob/main/docker/Dockerfile
# Build command:
# cd <sglang-repo>
# docker build -t orthozany/sglang-blackwell:luke-ar \
# -f docker/Dockerfile \
# --build-arg CUDA_VERSION=13.0.1 \
# --build-arg GRACE_BLACKWELL=1 \
# --build-arg BUILD_AND_DOWNLOAD_PARALLEL=8 \
# --build-arg USE_LATEST_SGLANG=0 \
# --build-arg BRANCH_TYPE=local \
# .
docker run --gpus all --ipc host --shm-size 32g \
-v /path/to/models/moonshotai/Kimi-K2.5:/model:ro \
-e NCCL_P2P_LEVEL=SYS -e NCCL_IB_DISABLE=1 -e NCCL_PROTO=LL \
-p 8400:8000 \
orthozany/sglang-blackwell:luke-ar \
python3 -m sglang.launch_server \
--model-path=/model \
--served-model-name=Kimi-K2.5 \
--tp=8 \
--host=0.0.0.0 --port=8000 \
--mem-fraction-static=0.94 \
--attention-backend=flashinfer \
--enable-pcie-oneshot-allreduce \
--enable-pcie-oneshot-allreduce-fusion \
--pcie-oneshot-allreduce-max-size=65536 \
--enable-torch-compile \
--enable-mixed-chunk \
--trust-remote-code \
--reasoning-parser=kimi_k2 \
--tool-call-parser=kimi_k2
| Flag | Description |
|---|---|
--enable-pcie-oneshot-allreduce | Enable PCIe system-scope barrier allreduce for small messages |
--enable-pcie-oneshot-allreduce-fusion | Fuse allreduce with RMSNorm for reduced kernel launches |
--pcie-oneshot-allreduce-max-size=65536 | Max message size (bytes) for custom AR (falls back to NCCL above this) |
| Context | Custom AR + TC | Baseline | Delta |
|---|---|---|---|
| Short | 116 | 106 | +9% |
| 4K | 108 | 99 | +9% |
| 8K | 94 | 87 | +8% |
| 16K | 86 | 80 | +8% |
| 32K | 80 | 75 | +7% |
| 128K | 63 | 60 | +5% |
| Concurrency | Aggregate tok/s |
|---|---|
| 8 | 469 |
| 16 | 720 |
| 64 | 1,085 |
| 128 | 1,852 |
Content type
Image
Digest
sha256:40b7d8306…
Size
11.3 GB
Last updated
8 months ago
docker pull orthozany/sglang-blackwell:luke-ar