Sign inSign up

sabreshao/sglang:glm_pcp_mori_tbo

Manifest digest

sha256:c640aecb5a87b4b3dd1998d5ae3ce4925544ab2db13dc4e0c088a52401449c8c

OS/ARCH

linux/amd64

Compressed size

26.52 GB

Last pushed

about 1 month by sabreshao

Type

Image

Manifest digest

sha256:c640aecb5a87b4b3dd1998d5ae3ce4925544ab2db13dc4e0c088a52401449c8c

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.ref.name=ubuntu0 B
4LABEL org.opencontainers.image.version=22.040 B
5ADD file ... in / 28.17 MB
6CMD ["/bin/bash"]0 B
7ARG PYTHON_VERSION0 B
8ARG PYTORCH_VERSION0 B
9ARG ROCM_VERSION0 B
10ARG AMDGPU_VERSION0 B
11ENV DEBIAN_FRONTEND=noninteractive0 B
12RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.2279.2 KB
13RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.2112.37 MB
14ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
15ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B
16RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.2102.09 MB
17RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.2121.99 MB
18ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
19RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.216.67 KB
20RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.2390.76 KB
21RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.27.61 GB
22RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.23.93 GB
23RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.9.1 ROCM_VERSION=7.2115.6 KB
24ENV BUILD_VLLM=00 B
25ENV BUILD_TRITON=10 B
26ENV BUILD_LLVM=00 B
27ENV BUILD_AITER_ALL=10 B
28ENV BUILD_MOONCAKE=10 B
29ENV AITER_COMMIT_DEFAULT=5bd9adf3d3c9ba8cf5b9e5fbc6826d2ce54688bf0 B
30ARG GPU_ARCH=gfx950-rocm7200 B
31ENV GPU_ARCH_LIST=gfx9500 B
32ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
33ARG SGL_REPO=https://github.com/sgl-project/sglang.git0 B
34ARG SGL_DEFAULT=main0 B
35ARG SGL_BRANCH=main0 B
36ARG BRANCH_TYPE=local0 B
37ARG SETUPTOOLS_SCM_PRETEND_VERSION=0 B
38ARG TRITON_REPO=https://github.com/triton-lang/triton.git0 B
39ARG TRITON_COMMIT=42270451990532c67e69d753fbd026f28fcc48400 B
40ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
41ARG AITER_COMMIT=0 B
42ENV AITER_COMMIT=5bd9adf3d3c9ba8cf5b9e5fbc6826d2ce54688bf0 B
43ARG LLVM_REPO=https://github.com/jrbyrnes/llvm-project.git0 B
44ARG LLVM_BRANCH=MainOpSelV20 B
45ARG LLVM_COMMIT=6520ace8227ffe2728148d5f3b9872a870b0a5600 B
46ARG MOONCAKE_REPO=https://github.com/kvcache-ai/Mooncake.git0 B
47ARG MOONCAKE_COMMIT=01d1eb2a7ec37fd5e20a88573e9b4956e7846e9a0 B
48ARG TILELANG_REPO=https://github.com/tile-ai/tilelang.git0 B
49ARG TILELANG_COMMIT=a55a82302bf7f3c5af635b5c9146f728185cc9000 B
50ARG FHT_REPO=https://github.com/jeffdaily/fast-hadamard-transform.git0 B
51ARG FHT_BRANCH=rocm0 B
52ARG FHT_COMMIT=46efb7d776d38638fc39f3c803eaee3dd7016bd10 B
53ARG ENABLE_MORI=10 B
54ARG NIC_BACKEND=none0 B
55ARG MORI_REPO=https://github.com/ROCm/mori.git0 B
56ARG MORI_COMMIT=f7e6ac6863c53821bc7afb91a578cc6ce38fcad00 B
57ARG ENABLE_NIXL=10 B
58ARG UCX_REPO=https://github.com/openucx/ucx.git0 B
59ARG UCX_BRANCH=v1.19.x0 B
60ARG NIXL_REPO=https://github.com/ai-dynamo/nixl.git0 B
61ARG NIXL_COMMIT=c28061f9782e099f975bcc79198b7b5a1a36cc400 B
62ARG AINIC_VERSION=1.117.5-a-380 B
63ARG UBUNTU_CODENAME=jammy0 B
64ARG UBUNTU_MIRROR=0 B
65USER root0 B
66RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main238 B
67RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B
68RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main5.12 MB
69RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main215 B
70RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main3.23 MB
71WORKDIR /sgl-workspace103 B
72RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B
73ENV SETUPTOOLS_SCM_PRETEND_VERSION=0 B
74ENV AITER_USE_SYSTEM_TRITON=10 B
75RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B
76RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main794.28 MB
77RUN |32 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main698.85 MB
78ARG RCCL_REPO=https://github.com/ROCm/rccl.git0 B
79ARG RCCL_BRANCH=29e1567b95e28823b0beb1a988adc587bfab5b4f0 B
80RUN |34 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main302.65 MB
81ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin0 B
82RUN |34 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main1.73 GB
83ARG BUILD_TYPE=all0 B
84ARG SETUPTOOLS_SCM_PRETEND_VERSION0 B
85RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main24.95 MB
86ENV PATH=/root/.cargo/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin0 B
87RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main290.35 MB
88ENV CARGO_BUILD_JOBS=40 B
89RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B
90COPY /src /tmp/local_src # buildkit185.28 MB
91RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main786.3 MB
92RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main20.99 KB
93RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B
94RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main1.42 GB
95ENV DEBIAN_FRONTEND=noninteractive0 B
96ENV LIBGL_ALWAYS_INDIRECT=10 B
97RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main214 B
98RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main1.07 GB
99RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main23.17 MB
100RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main6.77 MB
101RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main570.29 MB
102RUN |35 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main220.37 MB
103ARG TORCH_ROCM_FILE=torch-2.9.1+rocm7.2.0.lw.git7e1940d4-cp310-cp310-linux_x86_64.whl0 B
104RUN |36 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main925 B
105RUN |36 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main1.56 GB
106RUN |36 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main5 GB
107RUN |36 GPU_ARCH=gfx950-rocm720 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main9.48 KB
108ENV SGLANG_DISABLE_CUDNN_CHECK=10 B
109ENV HIP_FORCE_DEV_KERNARG=10 B
110ENV HSA_NO_SCRATCH_RECLAIM=10 B
111ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=10 B
112ENV SGLANG_INT4_WEIGHT=00 B
113ENV SGLANG_MOE_PADDING=10 B
114ENV SGLANG_ROCM_DISABLE_LINEARQUANT=00 B
115ENV SGLANG_ROCM_FUSED_DECODE_MLA=10 B
116ENV SGLANG_SET_CPU_AFFINITY=10 B
117ENV SGLANG_USE_AITER=10 B
118ENV SGLANG_USE_ROCM700A=10 B
119ENV NCCL_MIN_NCHANNELS=1120 B
120ENV ROCM_QUICK_REDUCE_QUANTIZATION=INT80 B
121ENV TORCHINDUCTOR_MAX_AUTOTUNE=10 B
122ENV TORCHINDUCTOR_MAX_AUTOTUNE_POINTWISE=10 B
123CMD ["/bin/bash"]0 B
124LABEL org.opencontainers.image.title=SGLang GLM-5.2 PD/MTP fixes0 B
125LABEL org.opencontainers.image.description=Validated PD warmup, sparse-DP,0 B
126COPY python/sglang/srt/layers/attention/dsa/dsa_indexer.py /sgl-workspace/sglang/python/sglang/srt/layers/attention/dsa/dsa_indexer.py # buildkit20.31 KB
127COPY python/sglang/srt/layers/attention/dsa_backend.py /sgl-workspace/sglang/python/sglang/srt/layers/attention/dsa_backend.py # buildkit29.42 KB
128COPY python/sglang/srt/speculative/eagle_utils.py /sgl-workspace/sglang/python/sglang/srt/speculative/eagle_utils.py # buildkit8.13 KB
129COPY python/sglang/srt/speculative/eagle_worker_v2.py /sgl-workspace/sglang/python/sglang/srt/speculative/eagle_worker_v2.py # buildkit15.81 KB
130COPY python/sglang/srt/speculative/spec_utils.py /sgl-workspace/sglang/python/sglang/srt/speculative/spec_utils.py # buildkit10.21 KB
131ENV PYTHONPATH=/sgl-workspace/sglang/python0 B
132LABEL org.opencontainers.image.title=SGLang GLM-5.2 FP8 (0807)0 B
133LABEL org.opencontainers.image.description=glm_0730 + local sglang/python0 B
134LABEL org.opencontainers.image.source=https://github.com/sgl-project/sglang0 B
135COPY sglang/python/ /sgl-workspace/sglang/python/ # buildkit13.87 MB
136COPY output/tuned_gemm/a8w8_blockscale_bpreshuffle_tuned_gemm.csv /sgl-workspace/aiter/aiter/configs/model_configs/glm5_a8w8_blockscale_bpreshuffle_tuned_gemm.csv # buildkit6.7 KB
137COPY output/tuned_gemm/bf16_tuned_gemm.csv /sgl-workspace/aiter/aiter/configs/model_configs/glm5_bf16_tuned_gemm.csv # buildkit4.84 KB
138ENV PYTHONPATH=/sgl-workspace/sglang/python:/sgl-workspace/aiter0 B
139ARG SGLANG_REVISION=2baaeef421180a9711717e23375f15302c3573d80 B
140ARG DSA_SOURCE_SHA256=34c3c683e373886a85e25e18ff28cabf697676fda1e2af305f35ed5bd314fc360 B
141LABEL org.opencontainers.image.title=GLM-5.2 FP8 PD real-row0 B
142LABEL org.opencontainers.image.base.name=qilihuan/sglang:glm_08070 B
143LABEL org.opencontainers.image.revision=2baaeef421180a9711717e23375f15302c3573d80 B
144LABEL org.opencontainers.image.dsa_source_sha256=34c3c683e373886a85e25e18ff28cabf697676fda1e2af305f35ed5bd314fc360 B
145COPY dsa_backend.py /sgl-workspace/sglang/python/sglang/srt/layers/attention/dsa_backend.py # buildkit33.03 KB
146ARG SGLANG_GIT_SHA=da4351bbf4d78b9083a032f6d72908ac6fbf76440 B
147ARG SGLANG_GIT_DIRTY=true0 B
148ARG SGLANG_BRANCH=glm5x-cp-tilelang-fp8kv0 B
149LABEL org.opencontainers.image.title=SGLang GLM-5.2 PCP +0 B
150COPY python/ /sgl-workspace/sglang/python/ # buildkit8.6 MB
151COPY MANIFEST.txt /sgl-workspace/sglang/PCP_MORI_TBO_MANIFEST.txt # buildkit583 B
152ENV PYTHONPATH=/sgl-workspace/sglang/python:/sgl-workspace/aiter0 B
153RUN |3 SGLANG_GIT_SHA=da4351bbf4d78b9083a032f6d72908ac6fbf7644 SGLANG_GIT_DIRTY=true SGLANG_BRANCH=glm5x-cp-tilelang-fp8kv4.43 KB

Command

 ARG RELEASE