rocm/atom-dev:nightly_202609161151-rocm10.0-yitingw1-opt_m3_dense_splits

Manifest digest

sha256:90f0939943ac25def2765df0d235935a2b6232cd0e4f9c29eb4f9526b018dcce

OS/ARCH

linux/amd64

Compressed size

13.96 GB

Last pushed

3 days by valarlip

Type

Image

Manifest digest

sha256:90f0939943ac25def2765df0d235935a2b6232cd0e4f9c29eb4f9526b018dcce

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;gfx9500 B

7ARG ROCM_SDK_VERSION=10.0.00 B

8ARG ROCM_TORCH_VERSION=2.11.00 B

9ARG ROCM_TORCHVISION_VERSION=0.26.00 B

10ARG ROCM_TORCHAUDIO_VERSION=2.11.00 B

11ARG ROCM_TRITON_VERSION=3.8.0+git4cff872c0 B

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

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

14RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0150.95 MB

15ENV VIRTUAL_ENV=/opt/venv0 B

16RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.04 MB

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

18RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.05.99 MB

19RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.04.2 GB

20RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.03.25 GB

21RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0278 B

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

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

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

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

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

27RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.01.22 KB

28RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0166 B

29RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0165 B

30RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.01.4 MB

31ENV PIP_CONSTRAINT=/etc/atom/constraints/torch-rocm.txt0 B

32RUN |7 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0215 B

33ARG GPU_ARCH=gfx942;gfx9500 B

34ARG BASE_IMAGE=rocm10-base0 B

35ENV GPU_ARCH_LIST=gfx942;gfx9500 B

36ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B

37ENV ATOM_BASE_IMAGE=rocm10-base0 B

38ENV HSA_ENABLE_IPC_MODE_LEGACY=10 B

39RUN |8 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0149.94 MB

40RUN |8 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.09.41 KB

41RUN |8 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0965.65 KB

42RUN |8 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.032 B

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

44ARG TRITON_PIN_VERSION=3.7.0+amd.rocm7.2.0.git890024100 B

45ARG TRITON_KERNELS_PIN_VERSION=1.0.0+amd.rocm7.2.0.git890024100 B

46RUN |11 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.032 B

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

48ARG ATOM_COMMIT=yitingw1/opt_m3_dense_splits0 B

49RUN |13 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0296.83 MB

50RUN |13 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.015.89 MB

51ARG INSTALL_MOONCAKE=10 B

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

53ARG MOONCAKE_COMMIT=v0.3.14-rc10 B

54ARG USE_HIP_DMABUF=ON0 B

55ARG VENV_PYTHON=/opt/venv/bin/python0 B

56RUN |18 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0117.72 MB

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

58RUN |18 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0212.53 MB

59

ARG IONIC_DEB_URL=https://repo.radeon.com/amdainic/pensando/ubuntu/1.117.1-a-63/pool/main/r/rdma-core/libionic1_54.0-149.g3304be71_amd64.deb

0 B

60RUN |19 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0304.3 KB

61RUN |19 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0239.55 MB

62ENV MOONCAKE_DISABLE_HIP_DMABUF=10 B

63ARG RUST_VERSION=1.94.00 B

64RUN |20 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0203.19 MB

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

66COPY /rccl-pkgs/ /tmp/rccl/ # buildkit24.01 MB

67RUN |20 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.025.32 MB

68COPY /app/aiter-test /app/aiter-test # buildkit1.89 GB

69RUN |20 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0997.58 MB

70RUN |20 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.092 B

71ARG CACHEBUST=17895546510 B

72RUN |21 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0733.68 MB

73RUN |21 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.032 B

74RUN |21 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.01.95 MB

75RUN |21 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.014.55 MB

76ARG LMCACHE_WHEEL_NAME=lmcache-0.5.5rc3+rocm7.2.4.torch2.10.git3d3aa833.cxx11abi1-cp312-cp312-manylinux_2_39_x86_64.whl0 B

77

ARG LMCACHE_WHEEL_URL=https://github.com/LMCache/LMCache/releases/download/v0.5.5rc3-rocm-torch210/lmcache-0.5.5rc3%2Brocm7.2.4.torch2.10.git3d3aa833.cxx11abi1-cp312-cp312-manylinux_2_39_x86_64.whl

0 B

78ARG LMCACHE_WHEEL_SHA256=06cda2fef1c2cf3926ffa59c4ba13b6029c6e40d3fba6db2bc2e7350a29c280f0 B

79ARG LMCACHE_TAG=v0.4.50 B

80RUN |25 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.01.21 GB

81ARG INSTALL_SA_AIPERF=10 B

82ARG SA_AIPERF_REF=754356e9a39acc6cc6afb242d123bb57c3fb6f750 B

83RUN |27 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0263.91 MB

84RUN |27 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.0166.91 KB

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

86ARG TRITON_PIN_VERSION=3.7.0+amd.rocm7.2.0.git890024100 B

87ARG TRITON_KERNELS_PIN_VERSION=1.0.0+amd.rocm7.2.0.git890024100 B

88RUN |27 GPU_ARCH=gfx942;gfx950 ROCM_SDK_VERSION=10.0.0 ROCM_TORCH_VERSION=2.11.032 B

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

Command

 ARG RELEASE