sabreshao/sglang:diffusion_1221

Manifest digest

sha256:5b59cb099c2c0eb1cd09296dbadfedaa972cf833c0e73ebbac2feb5434df5da9

OS/ARCH

linux/amd64

Compressed size

18.8 GB

Last pushed

9 months by sabreshao

Type

Image

Manifest digest

sha256:5b59cb099c2c0eb1cd09296dbadfedaa972cf833c0e73ebbac2feb5434df5da9

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 / 29.04 MB

6CMD ["/bin/bash"]0 B

7ARG PYTHON_VERSION=3.100 B

8ARG ROCM_BUILD_JOB=compute-rocm-rel-7.00 B

9ARG LKG_BUILD_NUM=380 B

10ARG ARTF_SERV=http://compute-artifactory.amd.com0 B

11ARG ARTF_PRE=http://compute-artifactory.amd.com/artifactory/list0 B

12ARG PYTORCH_VERSION=2.8.00 B

13ARG ROCM_VERSION0 B

14ARG PYTHON_VERSION_PYTORCH=3100 B

15ARG PY_POST_COMP=0 B

16

ARG WHL_INSTALL=http://compute-artifactory.amd.com/artifactory/compute-pytorch-rocm/compute-rocm-rel-7.0/38/torch-2.8.0+rocm7.0.0.lw.gitb2fb6885-cp310-cp310-linux_x86_64.sh

0 B

17ARG OS_VERSION=22.040 B

18ARG PYTORCH_ROCM_ARCH=gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1200;gfx1201;gfx950;gfx11510 B

19ENV DEBIAN_FRONTEND=noninteractive0 B

20RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38279.21 KB

21RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38109.38 MB

22ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

23ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B

24RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=3896.98 MB

25RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=387.91 MB

26ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

27RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=385.53 GB

28RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=381.52 MB

29COPY /pytorch/whl_docker/custom_install.sh /var/lib/jenkins/custom_install.sh # buildkit1.51 KB

30COPY src_repo /var/lib/jenkins/pytorch # buildkit1.1 GB

31RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38211 B

32ENV BUILD_ENVIRONMENT=rocm0 B

33ENV PYTORCH_ROCM_ARCH=gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1200;gfx1201;gfx950;gfx11510 B

34RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38778.27 MB

35RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=383.34 GB

36RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-rel-7.0 LKG_BUILD_NUM=38101 KB

37ENV PATH=/opt/rocm/llvm/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B

38ENV ROCM_PATH=/opt/rocm0 B

39ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:0 B

40ARG PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

41ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

42ARG PYTHON_VERSION=3.120 B

43RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh92 B

44WORKDIR /app32 B

45ENV DEBIAN_FRONTEND=noninteractive0 B

46RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh57.1 MB

47RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh63.73 MB

48RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh303.12 MB

49RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh32 B

50RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh1.02 GB

51

ARG BASE_IMAGE=registry-sc-harbor.amd.com/framework/compute-rocm-rel-7.0:38_ubuntu22.04_py3.10_pytorch_release-2.8_b2fb6885

0 B

52ARG HIPBLAS_COMMON_BRANCH0 B

53ARG HIPBLASLT_BRANCH0 B

54ARG LEGACY_HIPBLASLT_OPTION0 B

55ARG TRITON_BRANCH=shared/oai-oss-bf16mx4-moe-launch0 B

56ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B

57ARG PYTORCH_BRANCH=triton_kernel_wrap_fix0 B

58ARG PYTORCH_VISION_BRANCH=5dc9e7de0 B

59ARG PYTORCH_REPO=https://github.com/ROCm/pytorch.git0 B

60ARG PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git0 B

61ARG FA_BRANCH=3222ea30 B

62ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git0 B

63ARG AITER_BRANCH0 B

64ARG AITER_REPO0 B

65

RUN |16 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 BASE_IMAGE=registry-sc-harbor.amd.com/framework/compute-rocm-rel-7.0:38_ubuntu22.04_py3.10_pytorch_release-2.8_b2fb6885

388 B

66ARG ARG_PYTORCH_ROCM_ARCH0 B

67ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

68ENV AITER_ROCM_ARCH=gfx942;gfx9500 B

69RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c84.09 MB

70RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c1.34 KB

71RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c1.33 KB

72RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c32 B

73ARG COMMON_WORKDIR=/app0 B

74WORKDIR /app32 B

75RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh2.02 KB

76RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh32 B

77RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh9.16 MB

78RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh1.13 GB

79ARG COMMON_WORKDIR=/app0 B

80COPY /benchmarks /app/vllm/benchmarks # buildkit165.86 KB

81COPY /examples /app/vllm/examples # buildkit159.67 KB

82ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B

83ENV TOKENIZERS_PARALLELISM=false0 B

84ENV SAFETENSORS_FAST_GPU=10 B

85ENV HIP_FORCE_DEV_KERNARG=10 B

86CMD ["/bin/bash"]0 B

87/bin/bash1.05 KB

88ENV BUILD_VLLM=00 B

89ENV BUILD_TRITON=00 B

90ENV BUILD_LLVM=00 B

91ENV BUILD_AITER_ALL=00 B

92ENV BUILD_MOONCAKE=10 B

93ENV AITER_COMMIT=v0.1.7.post50 B

94ENV NO_DEPS_FLAG=0 B

95ARG GPU_ARCH=gfx9500 B

96ENV GPU_ARCH_LIST=gfx9500 B

97ARG SGL_REPO=https://github.com/sgl-project/sglang.git0 B

98ARG SGL_DEFAULT=main0 B

99ARG SGL_BRANCH=main0 B

100ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B

101ARG TRITON_COMMIT=improve_fa_decode_3.0.00 B

102ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B

103ARG LLVM_REPO=https://github.com/jrbyrnes/llvm-project.git0 B

104ARG LLVM_BRANCH=MainOpSelV20 B

105ARG LLVM_COMMIT=6520ace8227ffe2728148d5f3b9872a870b0a5600 B

106ARG MOONCAKE_REPO=https://github.com/kvcache-ai/Mooncake.git0 B

107ARG MOONCAKE_COMMIT=b6a841dc78c707ec655a563453277d969fb8f38d0 B

108ARG TILELANG_REPO=https://github.com/HaiShaw/tilelang.git0 B

109ARG TILELANG_BRANCH=dsv32-mi35x0 B

110ARG TILELANG_COMMIT=ae938cf885743f165a19656d1122ad42bb0e30b80 B

111ARG FHT_REPO=https://github.com/jeffdaily/fast-hadamard-transform.git0 B

112ARG FHT_BRANCH=rocm0 B

113ARG FHT_COMMIT=46efb7d776d38638fc39f3c803eaee3dd7016bd10 B

114USER root0 B

115RUN |18 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main4.59 MB

116RUN |18 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main197 B

117WORKDIR /sgl-workspace103 B

118RUN |18 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

119RUN |18 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main525 B

120RUN |18 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main487.88 MB

121RUN |18 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main795.49 KB

122RUN |18 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

123ARG VLLM_REPO=https://github.com/ROCm/vllm.git0 B

124ARG VLLM_BRANCH=9f6b92db47c3444b7a7d67451ba0c3a2d6af4c2c0 B

125RUN |20 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

126

ENV PATH=/opt/rocm/llvm/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin

0 B

127RUN |20 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main1.7 GB

128ARG BUILD_TYPE=all0 B

129RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main17.93 MB

130RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main32 B

131RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main389.98 MB

132RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main75.28 KB

133RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main2.75 KB

134

ENV PATH=/root/.cargo/bin:/opt/rocm/llvm/bin:/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin

0 B

135RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main284.36 MB

136RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main1.34 GB

137ENV DEBIAN_FRONTEND=noninteractive0 B

138ENV LIBGL_ALWAYS_INDIRECT=10 B

139RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main215 B

140RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main964.17 MB

141RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main22.72 MB

142RUN |21 GPU_ARCH=gfx950 SGL_REPO=https://github.com/sgl-project/sglang.git SGL_DEFAULT=main6.89 MB

143ENV HIP_FORCE_DEV_KERNARG=10 B

144ENV HSA_NO_SCRATCH_RECLAIM=10 B

145ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=10 B

146ENV SGLANG_INT4_WEIGHT=00 B

147ENV SGLANG_MOE_PADDING=10 B

148ENV SGLANG_ROCM_DISABLE_LINEARQUANT=00 B

149ENV SGLANG_ROCM_FUSED_DECODE_MLA=10 B

150ENV SGLANG_SET_CPU_AFFINITY=10 B

151ENV SGLANG_USE_AITER=10 B

152ENV SGLANG_USE_ROCM700A=10 B

153ENV NCCL_MIN_NCHANNELS=1120 B

154ENV VLLM_FP8_PADDING=10 B

155ENV VLLM_FP8_ACT_PADDING=10 B

156ENV VLLM_FP8_WEIGHT_PADDING=10 B

157ENV VLLM_FP8_REDUCE_CONV=10 B

158ENV TORCHINDUCTOR_MAX_AUTOTUNE=10 B

159ENV TORCHINDUCTOR_MAX_AUTOTUNE_POINTWISE=10 B

160CMD ["/bin/bash"]0 B

Command

 ARG RELEASE