rocm/sgl-dev:v0.5.21-rocm10-mi30x-20261015

Manifest digest

sha256:1e7092391f8c2d6b149197e98bc74e143c525a38e202c3e5aa6b23f4f773f897

OS/ARCH

linux/amd64

Compressed size

14.41 GB

Last pushed

29 days by henryx

Type

Image

Manifest digest

sha256:1e7092391f8c2d6b149197e98bc74e143c525a38e202c3e5aa6b23f4f773f897

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=gfx942-rocm10000 B

7ARG ROCM1000_SDK_VERSION=10.0.00 B

8ARG ROCM1000_TORCH_VERSION=2.11.00 B

9ARG ROCM1000_TORCHVISION_VERSION=0.26.00 B

10ARG ROCM1000_TORCHAUDIO_VERSION=2.11.00 B

11ARG ROCM1000_TRITON_VERSION=3.8.0+git4cff872c0 B

12ARG ROCM1010_SDK_VERSION=10.1.00 B

13ARG ROCM1010_TORCH_VERSION=2.12.00 B

14ARG ROCM1010_TORCHVISION_VERSION=0.27.00 B

15ARG ROCM1010_TORCHAUDIO_VERSION=2.11.00 B

16ARG ROCM1010_TRITON_VERSION=3.8.0+git669b31ac0 B

17ARG ROCM_INDEX_URL=https://stable.repo.amd.com/rocm/whl-next/0 B

18ARG ROCM_DEVICE_ARCH_LIST=gfx942 gfx950 gfx12500 B

19RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0153.35 MB

20ENV VIRTUAL_ENV=/opt/venv0 B

21RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.04 MB

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

23RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.06 MB

24RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.03.41 GB

25RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.02.6 GB

26RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0277 B

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

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

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

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

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

32RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.22 KB

33RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0168 B

34RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0166 B

35ENV BUILD_VLLM=00 B

36ENV BUILD_TRITON=00 B

37ENV BUILD_LLVM=00 B

38ENV BUILD_AITER_ALL=10 B

39ENV BUILD_MOONCAKE=10 B

40ENV AITER_COMMIT_DEFAULT=2bb549102ba2d9e51da8ef21107361827075d8b60 B

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

42RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0174 B

43ARG GPU_ARCH=gfx942-rocm10000 B

44RUN |13 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.3 KB

45ENV GPU_ARCH=gfx942-rocm10000 B

46ENV GPU_ARCH_LIST=gfx9420 B

47ENV PYTORCH_ROCM_ARCH=gfx942;gfx950;gfx12500 B

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

49ARG SGL_DEFAULT=main0 B

50ARG SGL_BRANCH=bba535ebb4d176c6bf8361619e0c29fa263642fe0 B

51ARG BRANCH_TYPE=remote0 B

52ARG SETUPTOOLS_SCM_PRETEND_VERSION=0.5.21.dev20261015+gbba535ebb40 B

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

54ENV TRITON_COMMIT=0 B

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

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

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

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

59ARG TORCH_ROCM_VERSION=2.11.0+rocm7.20 B

60ARG TORCHVISION_ROCM_VERSION=0.26.0+rocm7.20 B

61ARG TORCHAUDIO_ROCM_VERSION=2.11.0+rocm7.20 B

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

63ARG AITER_COMMIT=0 B

64ENV AITER_COMMIT=2bb549102ba2d9e51da8ef21107361827075d8b60 B

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

66ARG LLVM_BRANCH=MainOpSelV20 B

67ARG LLVM_COMMIT=6520ace8227ffe2728148d5f3b9872a870b0a5600 B

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

69ARG MOONCAKE_COMMIT=01d1eb2a7ec37fd5e20a88573e9b4956e7846e9a0 B

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

71ARG TILELANG_COMMIT=a55a82302bf7f3c5af635b5c9146f728185cc9000 B

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

73ARG FHT_BRANCH=rocm0 B

74ARG FHT_COMMIT=46efb7d776d38638fc39f3c803eaee3dd7016bd10 B

75ARG ENABLE_MORI=10 B

76ARG NIC_BACKEND=none0 B

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

78ARG MORI_COMMIT=f39fa765e897c95995957d0518908cc982a88fc20 B

79ARG ENABLE_NIXL=10 B

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

81ARG UCX_BRANCH=v1.19.x0 B

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

83ARG NIXL_COMMIT=c28061f9782e099f975bcc79198b7b5a1a36cc400 B

84ARG AINIC_VERSION=1.117.5-a-380 B

85ARG UBUNTU_CODENAME=jammy0 B

86ARG UBUNTU_MIRROR=https://archive.ubuntu.com0 B

87USER root0 B

88RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.34 KB

89RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

90RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.039.26 KB

91RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

92RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.24 MB

93RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

94RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0220 B

95WORKDIR /sgl-workspace103 B

96RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

97ENV SETUPTOOLS_SCM_PRETEND_VERSION=0 B

98ENV AITER_USE_SYSTEM_TRITON=10 B

99RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

100RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0733.6 MB

101

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

102RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0586.83 MB

103RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.83 KB

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

105RUN |50 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.76 GB

106ARG BUILD_TYPE=all0 B

107ARG SETUPTOOLS_SCM_PRETEND_VERSION=0.5.21.dev20261015+gbba535ebb40 B

108RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.014.66 MB

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

110RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0295.11 MB

111ENV CARGO_BUILD_JOBS=40 B

112RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

113COPY /src /tmp/local_src # buildkit464.26 MB

114RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0482.77 MB

115RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0197 B

116RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0821.7 MB

117RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0175 B

118RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

119RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

120RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.34 GB

121ENV DEBIAN_FRONTEND=noninteractive0 B

122ENV LIBGL_ALWAYS_INDIRECT=10 B

123RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0235 B

124RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01006.68 MB

125RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.017.74 MB

126RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.06.85 MB

127RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0584.8 MB

128RUN |51 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0217.09 MB

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

130RUN |52 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0925 B

131RUN |52 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

132RUN |52 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

133RUN |52 GPU_ARCH=gfx942-rocm1000 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

134ENV SGLANG_DISABLE_CUDNN_CHECK=10 B

135ENV HIP_FORCE_DEV_KERNARG=10 B

136ENV HSA_NO_SCRATCH_RECLAIM=10 B

137ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=10 B

138ENV SGLANG_INT4_WEIGHT=00 B

139ENV SGLANG_MOE_PADDING=10 B

140ENV SGLANG_ROCM_DISABLE_LINEARQUANT=00 B

141ENV SGLANG_ROCM_FUSED_DECODE_MLA=10 B

142ENV SGLANG_SET_CPU_AFFINITY=10 B

143ENV SGLANG_USE_AITER=10 B

144ENV SGLANG_USE_ROCM700A=10 B

145ENV NCCL_MIN_NCHANNELS=1120 B

146ENV ROCM_QUICK_REDUCE_QUANTIZATION=INT80 B

147ENV TORCHINDUCTOR_MAX_AUTOTUNE=10 B

148ENV TORCHINDUCTOR_MAX_AUTOTUNE_POINTWISE=10 B

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

Command

 ARG RELEASE