sabreshao/sglang:rocm-10.2-nightly-gfx950

Manifest digest

sha256:35215b48ea65b22ceb7de6d58d7904afbc16d03643cdac4c0451b68811c4b2db

OS/ARCH

linux/amd64

Compressed size

12.86 GB

Last pushed

about 12 hours by sabreshao

Type

Image

Manifest digest

sha256:35215b48ea65b22ceb7de6d58d7904afbc16d03643cdac4c0451b68811c4b2db

Image Layers

Image Layers

1ARG RELEASE0 B

2ARG LAUNCHPAD_BUILD_ARCH0 B

3LABEL org.opencontainers.image.version=24.040 B

4ADD file ... in / 28.39 MB

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

6ARG GPU_ARCH=gfx950-rocm1020 B

7ARG ROCM_NIGHTLY_VERSION=10.2.0a202610110 B

8ARG ROCM_NIGHTLY_TORCH_VERSION=2.12.00 B

9ARG ROCM_NIGHTLY_TORCHVISION_VERSION=0.27.00 B

10ARG ROCM_NIGHTLY_TORCHAUDIO_VERSION=2.11.00 B

11ARG ROCM_NIGHTLY_TRITON_VERSION=3.8.0+git669b31ac0 B

12ARG ROCM_NIGHTLY_SDK_INDEX_URL=https://nightly.repo.amd.com/rocm/whl-next/0 B

13ARG ROCM_NIGHTLY_PYTORCH_INDEX_URL=https://nightly.repo.amd.com/rocm/pytorch/whl-next/0 B

14ARG ROCM_DEVICE_ARCH_LIST=gfx942 gfx9500 B

15RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0153.29 MB

16ENV VIRTUAL_ENV=/opt/venv0 B

17RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.04 MB

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

19RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.06 MB

20RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.02.91 GB

21RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.02.07 GB

22RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0278 B

23ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

24ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

25ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B

26ENV LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B

27ENV LD_LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B

28RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.01.22 KB

29RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0168 B

30RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0166 B

31ENV BUILD_VLLM=00 B

32ENV BUILD_TRITON=00 B

33ENV BUILD_LLVM=00 B

34ENV BUILD_AITER_ALL=10 B

35ENV BUILD_MOONCAKE=10 B

36ENV AITER_COMMIT_DEFAULT=e7d2453f25e5aaeea72a3a40b284ba453e0266240 B

37ENV PIP_CONSTRAINT=/etc/sglang/constraints/torch-rocm.txt0 B

38RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0174 B

39ARG GPU_ARCH=gfx950-rocm1020 B

40RUN |9 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.01.3 KB

41ENV GPU_ARCH=gfx950-rocm1020 B

42ENV GPU_ARCH_LIST=gfx9500 B

43ENV PYTORCH_ROCM_ARCH=gfx942;gfx950;gfx12500 B

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

45ARG SGL_DEFAULT=main0 B

46ARG SGL_BRANCH=main0 B

47ARG BRANCH_TYPE=remote0 B

48ARG SETUPTOOLS_SCM_PRETEND_VERSION=0 B

49ARG TRITON_REPO=https://github.com/triton-lang/triton.git0 B

50ENV TRITON_COMMIT=0 B

51ARG TRITON_INDEX_URL=https://pypi.amd.com/triton/release/rocm-7.2.0/simple/0 B

52ARG TRITON_VERSION=3.7.0+amd.rocm7.2.0.git890024100 B

53ARG TRITON_KERNELS_VERSION=1.0.0+amd.rocm7.2.0.git890024100 B

54ARG TORCH_ROCM_INDEX_URL=https://download.pytorch.org/whl/rocm7.20 B

55ARG TORCH_ROCM_VERSION=2.11.0+rocm7.20 B

56ARG TORCHVISION_ROCM_VERSION=0.26.0+rocm7.20 B

57ARG TORCHAUDIO_ROCM_VERSION=2.11.0+rocm7.20 B

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

59ARG AITER_COMMIT=0 B

60ENV AITER_COMMIT=e7d2453f25e5aaeea72a3a40b284ba453e0266240 B

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

62ARG LLVM_BRANCH=MainOpSelV20 B

63ARG LLVM_COMMIT=6520ace8227ffe2728148d5f3b9872a870b0a5600 B

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

65ARG MOONCAKE_COMMIT=01d1eb2a7ec37fd5e20a88573e9b4956e7846e9a0 B

66ARG TILELANG_REPO=https://github.com/tile-ai/tilelang.git0 B

67ARG TILELANG_COMMIT=a55a82302bf7f3c5af635b5c9146f728185cc9000 B

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

69ARG FHT_BRANCH=rocm0 B

70ARG FHT_COMMIT=46efb7d776d38638fc39f3c803eaee3dd7016bd10 B

71ARG ENABLE_MORI=00 B

72ARG NIC_BACKEND=none0 B

73ARG MORI_REPO=https://github.com/ROCm/mori.git0 B

74ARG MORI_COMMIT=879983bdbd8c65c52e9f79ad836a61cbffef98b60 B

75ARG ENABLE_NIXL=10 B

76ARG UCX_REPO=https://github.com/openucx/ucx.git0 B

77ARG UCX_BRANCH=v1.19.x0 B

78ARG NIXL_REPO=https://github.com/ai-dynamo/nixl.git0 B

79ARG NIXL_COMMIT=c28061f9782e099f975bcc79198b7b5a1a36cc400 B

80ARG AINIC_VERSION=1.117.5-a-380 B

81ARG UBUNTU_CODENAME=jammy0 B

82ARG UBUNTU_MIRROR=0 B

83USER root0 B

84RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0234 B

85RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

86RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.039.32 KB

87RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0192 B

88RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0170.79 KB

89RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

90RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0230 B

91WORKDIR /sgl-workspace103 B

92RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

93ENV SETUPTOOLS_SCM_PRETEND_VERSION=0 B

94ENV AITER_USE_SYSTEM_TRITON=10 B

95RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

96RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0743.7 MB

97

COPY /src/python/sglang/kernels/ops/attention/dsv4/asm/gfx950/mla_v4/mla_a8w8_qh64_qseqlen1_gqaratio64_nm.co /sgl-workspace/aiter/hsa/gfx950/mla_v4/ # buildkit

16.91 KB

98RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0651.02 MB

99RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.01.83 KB

100ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin0 B

101RUN |46 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.01.76 GB

102ARG BUILD_TYPE=all0 B

103ARG SETUPTOOLS_SCM_PRETEND_VERSION0 B

104RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.014.65 MB

105ENV PATH=/root/.cargo/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/go/bin0 B

106RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0295.17 MB

107ENV CARGO_BUILD_JOBS=40 B

108RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

109COPY /src /tmp/local_src # buildkit451.65 MB

110RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0471.49 MB

111RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0206 B

112RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0823.93 MB

113RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0177 B

114RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.092 B

115RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

116RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.01.34 GB

117ENV DEBIAN_FRONTEND=noninteractive0 B

118ENV LIBGL_ALWAYS_INDIRECT=10 B

119RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0237 B

120RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.01004.88 MB

121RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.017.56 MB

122RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.06.85 MB

123RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

124RUN |47 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0215.7 MB

125ARG TORCH_ROCM_FILE=torch-2.9.1+rocm7.2.0.lw.git7e1940d4-cp310-cp310-linux_x86_64.whl0 B

126RUN |48 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.0926 B

127RUN |48 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

128RUN |48 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

129RUN |48 GPU_ARCH=gfx950-rocm102 ROCM_NIGHTLY_VERSION=10.2.0a20261011 ROCM_NIGHTLY_TORCH_VERSION=2.12.032 B

130ENV SGLANG_DISABLE_CUDNN_CHECK=10 B

131ENV HIP_FORCE_DEV_KERNARG=10 B

132ENV HSA_NO_SCRATCH_RECLAIM=10 B

133ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=10 B

134ENV SGLANG_INT4_WEIGHT=00 B

135ENV SGLANG_MOE_PADDING=10 B

136ENV SGLANG_ROCM_DISABLE_LINEARQUANT=00 B

137ENV SGLANG_ROCM_FUSED_DECODE_MLA=10 B

138ENV SGLANG_SET_CPU_AFFINITY=10 B

139ENV SGLANG_USE_AITER=10 B

140ENV SGLANG_USE_ROCM700A=10 B

141ENV NCCL_MIN_NCHANNELS=1120 B

142ENV ROCM_QUICK_REDUCE_QUANTIZATION=INT80 B

143ENV TORCHINDUCTOR_MAX_AUTOTUNE=10 B

144ENV TORCHINDUCTOR_MAX_AUTOTUNE_POINTWISE=10 B

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

Command

 ARG RELEASE