rocm/atom:nightly_rocm_7.2_preview_202601260317

Manifest digest

sha256:a56abb575f3083b907eb8fb78c22d068c6ad27b3ff25895866a6242a7d68bc26

OS/ARCH

linux/amd64

Compressed size

24.75 GB

Last pushed

6 months by valarlip

Type

Image

Manifest digest

sha256:a56abb575f3083b907eb8fb78c22d068c6ad27b3ff25895866a6242a7d68bc26

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 ROCM_BUILD_JOB0 B

9ARG LKG_BUILD_NUM0 B

10ARG ARTF_SERV0 B

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

12ARG PYTORCH_VERSION0 B

13ARG ROCM_VERSION0 B

14ARG PYTHON_VERSION_PYTORCH0 B

15ARG PY_POST_COMP0 B

16ARG WHL_INSTALL0 B

17ARG OS_VERSION0 B

18ARG PYTORCH_ROCM_ARCH0 B

19ENV DEBIAN_FRONTEND=noninteractive0 B

20RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=110279.21 KB

21RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=110111.66 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-dkms-component-baas-rel LKG_BUILD_NUM=11097.01 MB

25RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=1107.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-dkms-component-baas-rel LKG_BUILD_NUM=1106.24 GB

28RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=1102.07 MB

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

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

31RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=110211 B

32ENV BUILD_ENVIRONMENT=rocm0 B

33ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

34RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=110761.65 MB

35RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=1101.82 GB

36RUN |12 PYTHON_VERSION=3.10 ROCM_BUILD_JOB=compute-rocm-dkms-component-baas-rel LKG_BUILD_NUM=110111.67 KB

37ENV PATH=/opt/rocm/llvm/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/sh93 B

44WORKDIR /app32 B

45ENV DEBIAN_FRONTEND=noninteractive0 B

46RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh64.84 MB

47RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh64.36 MB

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

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/sh32 B

51RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh345.49 MB

52RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh1.14 MB

53RUN |2 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 /bin/sh1.08 GB

54ARG BASE_IMAGE=rocm/vllm-private:110_ubuntu22.04_py3.10_pytorch_rocm7.1_internal_testing_dd9431140 B

55ARG HIPBLAS_COMMON_BRANCH0 B

56ARG HIPBLASLT_BRANCH=9176baa4996408fd323ed7c1a1492a09997e8d050 B

57ARG LEGACY_HIPBLASLT_OPTION0 B

58ARG TRITON_BRANCH=57c693b627fe058878ade4163a0a8df95d9fefa10 B

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

60ARG PYTORCH_BRANCH=rocm7.1_internal_testing-tritonlatest0 B

61ARG PYTORCH_VISION_BRANCH=5dc9e7de0 B

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

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

64ARG FA_BRANCH=3222ea30 B

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

66ARG AITER_BRANCH0 B

67ARG AITER_REPO0 B

68

RUN |16 PYTORCH_ROCM_ARCH=gfx942;gfx950 PYTHON_VERSION=3.12 BASE_IMAGE=rocm/vllm-private:110_ubuntu22.04_py3.10_pytorch_rocm7.1_internal_testing_dd943114

430 B

69ARG ARG_PYTORCH_ROCM_ARCH0 B

70ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

71ENV AITER_ROCM_ARCH=gfx942;gfx9500 B

72RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c101.77 MB

73RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c1.37 KB

74RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c1.37 KB

75RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c217 B

76ARG COMMON_WORKDIR=/app0 B

77WORKDIR /app32 B

78RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh2.07 KB

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

80RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh12.09 MB

81RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh1.5 GB

82RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh20.6 MB

83ARG COMMON_WORKDIR=/app0 B

84COPY /benchmarks /app/vllm/benchmarks # buildkit170.47 KB

85COPY /examples /app/vllm/examples # buildkit168.62 KB

86COPY /docker /app/vllm/docker # buildkit2.29 KB

87ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B

88ENV RAY_EXPERIMENTAL_NOSET_HIP_VISIBLE_DEVICES=10 B

89ENV TOKENIZERS_PARALLELISM=false0 B

90ENV VLLM_ROCM_USE_AITER=10 B

91ENV SAFETENSORS_FAST_GPU=10 B

92ENV HIP_FORCE_DEV_KERNARG=10 B

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

94/bin/bash847.25 MB

95/bin/bash1.1 GB

96/bin/bash1.83 GB

97/bin/bash612.13 MB

98ARG GPU_ARCH=gfx942;gfx9500 B

99ENV GPU_ARCH_LIST=gfx942;gfx9500 B

100ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

101ARG ATOM_REPO=https://github.com/ROCm/ATOM.git0 B

102ARG ATOM_COMMIT=HEAD0 B

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

104ARG AITER_COMMIT=HEAD0 B

105ARG MORI_COMMIT=b0dce4beebeb1f26c784eee17d5fd9785ee9447f0 B

106ARG PREBUILD_KERNELS=10 B

107ARG MAX_JOBS=640 B

108RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD1.53 KB

109RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD92 B

110RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

111RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD47.22 MB

112RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD44.38 MB

113RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD203.19 MB

114RUN |8 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

115ARG RCCL_REPO=https://github.com/ROCm/rccl.git0 B

116ARG RCCL_BRANCH=29e1567b95e28823b0beb1a988adc587bfab5b4f0 B

117ARG RCCL_SRC_DIR=/app/rccl/0 B

118RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD92 B

119RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD464.24 MB

120RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD20.62 MB

121RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

122RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD320 B

123RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD4.95 GB

124RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

125RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

126RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD388 B

127RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD1.23 GB

128RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

129RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD381 B

130RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD1.54 MB

131RUN |11 GPU_ARCH=gfx942;gfx950 ATOM_REPO=https://github.com/ROCm/ATOM.git ATOM_COMMIT=HEAD32 B

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

Command

 ARG RELEASE