rocm/sgl-dev:v0.5.23-rocm101-mi45x-20261102_rccl_d222bf_test

Manifest digest

sha256:60dd25fdeec41f592cafeeeac895d6bc30c71161ab6aa035e80249d8599e0064

OS/ARCH

linux/amd64

Compressed size

17.81 GB

Last pushed

1 day by henryx

Type

Image

Manifest digest

sha256:60dd25fdeec41f592cafeeeac895d6bc30c71161ab6aa035e80249d8599e0064

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=gfx1250-rocm10100 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=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0151.05 MB

20ENV VIRTUAL_ENV=/opt/venv0 B

21RUN |13 GPU_ARCH=gfx1250-rocm1010 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=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.06.14 MB

24RUN |13 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.02.52 GB

25RUN |13 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.79 GB

26RUN |13 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 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=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.22 KB

33RUN |13 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0167 B

34RUN |13 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0165 B

35ENV BUILD_VLLM=00 B

36ENV BUILD_TRITON=10 B

37ENV TRITON_COMMIT_DEFAULT=76940ad348795521b3dc9f6c79acd7309ff924e30 B

38ENV TRITON_REVERT_COMMITS_DEFAULT=9debefe44117a44c274b05033cfe9e4858883b860 B

39ENV BUILD_LLVM=00 B

40ENV BUILD_AITER_ALL=10 B

41ENV BUILD_MOONCAKE=10 B

42ENV AITER_COMMIT_DEFAULT=26f4f1fb2de14d122c71f5fed7afb97a5c97cade0 B

43ENV BUILD_RCCL=10 B

44ENV RCCL_COMMIT_DEFAULT=d2e0de357464cd34cabaea859b71c10da88e7ae90 B

45ENV HIPBLASLT_TENSILE_LIBPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_libraries/lib/hipblaslt/library/gfx12500 B

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

47RUN |13 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0172 B

48ARG GPU_ARCH=gfx1250-rocm10100 B

49RUN |13 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.3 KB

50ENV GPU_ARCH=gfx1250-rocm10100 B

51ENV GPU_ARCH_LIST=gfx12500 B

52ENV PYTORCH_ROCM_ARCH=gfx942;gfx950;gfx12500 B

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

54ARG SGL_DEFAULT=main0 B

55ARG SGL_BRANCH=c1f56c64cd91acefa95fab4c6dd88f7ca54fda790 B

56ARG BRANCH_TYPE=remote0 B

57ARG SETUPTOOLS_SCM_PRETEND_VERSION=0.5.23.dev20261102+gc1f56c64cd0 B

58ARG TRITON_REPO=0 B

59ENV TRITON_REPO=https://github.com/triton-lang/triton.git0 B

60ENV TRITON_COMMIT=76940ad348795521b3dc9f6c79acd7309ff924e30 B

61ARG TRITON_REVERT_COMMITS=0 B

62ENV TRITON_REVERT_COMMITS=9debefe44117a44c274b05033cfe9e4858883b860 B

63ARG BUILD_UTLX=00 B

64ENV BUILD_UTLX=00 B

65RUN |21 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

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

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

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

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

70ARG TORCH_ROCM_VERSION=2.11.0+rocm7.20 B

71ARG TORCHVISION_ROCM_VERSION=0.26.0+rocm7.20 B

72ARG TORCHAUDIO_ROCM_VERSION=2.11.0+rocm7.20 B

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

74ARG AITER_COMMIT=0 B

75ENV AITER_COMMIT=26f4f1fb2de14d122c71f5fed7afb97a5c97cade0 B

76ARG RCCL_REPO=https://github.com/ROCm/rocm-systems.git0 B

77ARG RCCL_COMMIT=0 B

78ENV RCCL_COMMIT=d2e0de357464cd34cabaea859b71c10da88e7ae90 B

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

80ARG LLVM_BRANCH=MainOpSelV20 B

81ARG LLVM_COMMIT=6520ace8227ffe2728148d5f3b9872a870b0a5600 B

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

83ARG MOONCAKE_COMMIT=78566d459c140956fd15606e102132b1acc83eec0 B

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

85ARG TILELANG_COMMIT=a55a82302bf7f3c5af635b5c9146f728185cc9000 B

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

87ARG FHT_BRANCH=rocm0 B

88ARG FHT_COMMIT=46efb7d776d38638fc39f3c803eaee3dd7016bd10 B

89ARG ENABLE_SGL_AMD_MONO=10 B

90ARG SGL_AMD_MONO_REPO=https://github.com/sgl-project/sgl-amd-mono.git0 B

91ARG SGL_AMD_MONO_COMMIT=0 B

92ARG ENABLE_MORI=10 B

93ARG NIC_BACKEND=none0 B

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

95ARG MORI_COMMIT=f39fa765e897c95995957d0518908cc982a88fc20 B

96ARG ENABLE_NIXL=10 B

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

98ARG UCX_BRANCH=v1.19.x0 B

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

100ARG NIXL_COMMIT=c28061f9782e099f975bcc79198b7b5a1a36cc400 B

101ARG AINIC_VERSION=1.117.5-a-380 B

102ARG UBUNTU_CODENAME=jammy0 B

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

104USER root0 B

105RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.34 KB

106RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

107RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.039.53 KB

108RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

109RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.22 MB

110RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

111RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0223 B

112WORKDIR /sgl-workspace103 B

113RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

114ENV SETUPTOOLS_SCM_PRETEND_VERSION=0 B

115ENV AITER_USE_SYSTEM_TRITON=10 B

116RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

117RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0834.52 MB

118

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

119RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0804.86 KB

120RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.82 KB

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

122RUN |57 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.98 GB

123ARG BUILD_TYPE=all0 B

124ARG SETUPTOOLS_SCM_PRETEND_VERSION=0.5.23.dev20261102+gc1f56c64cd0 B

125RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.014.73 MB

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

127RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0295.12 MB

128ENV CARGO_BUILD_JOBS=40 B

129RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

130COPY /src /tmp/local_src # buildkit539.29 MB

131RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0550.07 MB

132RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0201 B

133RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0840.1 MB

134RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0176 B

135RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0130 B

136RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

137RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01.33 GB

138ENV DEBIAN_FRONTEND=noninteractive0 B

139ENV LIBGL_ALWAYS_INDIRECT=10 B

140RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0237 B

141RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.01022.56 MB

142RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.019.71 MB

143RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

144RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.06.97 MB

145RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0596.89 MB

146RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0218.2 MB

147RUN |58 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0112.01 MB

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

149RUN |59 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.0925 B

150RUN |59 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

151RUN |59 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.05.07 GB

152RUN |59 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.012.24 KB

153ARG UTLX_REPO=https://github.com/triton-lang/triton-ext.git0 B

154ARG UTLX_COMMIT=268252bcc64e13b553b7906e5b69f02fa69054fb0 B

155RUN |61 GPU_ARCH=gfx1250-rocm1010 ROCM1000_SDK_VERSION=10.0.0 ROCM1000_TORCH_VERSION=2.11.032 B

156ENV SGLANG_DISABLE_CUDNN_CHECK=10 B

157ENV HIP_FORCE_DEV_KERNARG=10 B

158ENV HSA_NO_SCRATCH_RECLAIM=10 B

159ENV SGLANG_ALLOW_OVERWRITE_LONGER_CONTEXT_LEN=10 B

160ENV SGLANG_INT4_WEIGHT=00 B

161ENV SGLANG_MOE_PADDING=10 B

162ENV SGLANG_ROCM_DISABLE_LINEARQUANT=00 B

163ENV SGLANG_ROCM_FUSED_DECODE_MLA=10 B

164ENV SGLANG_SET_CPU_AFFINITY=10 B

165ENV SGLANG_USE_AITER=10 B

166ENV SGLANG_USE_ROCM700A=10 B

167ENV NCCL_MIN_NCHANNELS=1120 B

168ENV TORCHINDUCTOR_MAX_AUTOTUNE=10 B

169ENV TORCHINDUCTOR_MAX_AUTOTUNE_POINTWISE=10 B

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

Command

 ARG RELEASE