Sign inSign up

rocm/primus:v26.5.1

Index digest

sha256:0852eaa470c6e09323837438a15170da9b09e2f4f8bc916c9ebbf94429f0f564

OS/ARCH

linux/amd64

Compressed size

13.99 GB

Last pushed

1 day by rocm

Type

Image

Manifest digest

sha256:f2e3e66c2c1f8449085af614eaeed392a526c8fbcd6ec1c4991324186a33e4b1

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 28.37 MB
5CMD ["/bin/bash"]0 B
6WORKDIR /workspace/99 B
7ENV MAX_JOBS=1280 B
8ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
9ENV ROCM_AMDGPU_TARGETS=gfx942,gfx9500 B
10ENV DEBIAN_FRONTEND=noninteractive0 B
11RUN /bin/sh -c apt update274.36 MB
12RUN /bin/sh -c python -m4 MB
13ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
14RUN /bin/sh -c pip install33.35 MB
15RUN /bin/sh -c apt update137.53 MB
16ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
17ENV HSA_NO_SCRATCH_RECLAIM=10 B
18ENV THE_ROCK_VERSION=7.15.0a202607270 B
19RUN /bin/sh -c python -m3.3 GB
20ARG PYTORCH_VERSION=2.12.0+rocm7.15.0a202607270 B
21RUN |1 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 /bin/sh -c73.67 MB
22RUN |1 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 /bin/sh -c980 MB
23RUN |1 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 /bin/sh -c3.29 GB
24ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
25ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
26ENV HIP_PLATFORM=amd0 B
27ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
28ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
29ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
30ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
31ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
32ENV PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
33ENV LD_LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib64:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/host-math/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib0 B
34ENV LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib640 B
35ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
36ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
37RUN |1 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 /bin/sh -c23.88 KB
38ARG FA_REPO=https://github.com/ROCm/flash-attention.git0 B
39ARG FA_BRANCH=6387433156558135a998d5568a9d74c1778666d80 B
40ENV GPU_ARCHS=gfx942;gfx9500 B
41RUN |3 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d873.02 MB
42ARG TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf80 B
43ENV NVTE_USE_CAST_TRANSPOSE_TRITON=10 B
44ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B
45ENV NVTE_CK_USES_BWD_V3=10 B
46ENV NVTE_CK_USES_FWD_V3=10 B
47ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B
48ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
49RUN |4 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d821.64 MB
50RUN |4 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8322.23 MB
51ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
52ARG FLUX_BRANCH=26.4_RC_Patch_torchcodec0 B
53RUN |6 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8105.58 MB
54ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B
55ARG DLRM_BRANCH=main0 B
56RUN |8 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8103.58 KB
57ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
58ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
59RUN |10 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d821.85 MB
60ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
61ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B
62RUN |12 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d83.61 MB
63RUN |12 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8210.74 MB
64ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
65ARG GROUPED_GEMM_BRANCH=rocm0 B
66RUN |14 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8112.55 KB
67ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B
68ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B
69ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B
70ENV MAMBA_FORCE_BUILD=TRUE0 B
71ENV HIP_ARCHITECTURES=gfx942,gfx9500 B
72RUN |16 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d89.16 MB
73ARG MAMBA_REPO=https://github.com/AndreasKaratzas/mamba.git0 B
74ARG MAMBA_BRANCH=enable-primus-hybrid-models0 B
75RUN |18 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8277.59 MB
76ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
77ARG PRIMUS_BRANCH=9a86ba8c99d6d41199442edef6d02a2bfad71c7c0 B
78ENV NVTE_FLASH_ATTN=00 B
79ENV NVTE_FUSED_ATTN=10 B
80RUN |20 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d81.31 GB
81ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
82ARG AITER_COMMIT=0f3c58e6edb6754940bcf9fd5f09ccb6f389f52e0 B
83RUN |22 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8226.86 MB
84ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
85ARG TURBO_COMMIT=edc8d2ccb0be4888e80ee7c6e765fd3956026a320 B
86ENV HCC_AMDGPU_TARGET=gfx942,gfx9500 B
87RUN |24 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d81.11 GB
88RUN |24 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d813.32 MB
89WORKDIR /workspace/32 B
90RUN |24 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8424.79 MB
91ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B
92ARG FBGEMM_COMMIT=80bd3c077dc41b55cd16ed4dcad15cf7c1c1d76a0 B
93ENV BUILD_ROCM_VERSION=7.140 B
94RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8975.87 KB
95RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d81.71 GB
96RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8378.03 KB
97ENV GPU_ARCHS=native0 B
98RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d88.08 MB
99RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8514 B
100RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d811.07 MB
101ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
102RUN |27 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d82.74 MB
103RUN |27 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8229 B
104ARG UCX_VERSION=1.18.00 B
105RUN |28 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d886.43 MB
106ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
107ARG MPI_VERSION=4.1.60 B
108RUN |29 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d83.84 MB
109ENV PATH=/opt/openmpi/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
110ENV LD_LIBRARY_PATH=/opt/openmpi/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib64:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/host-math/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib0 B
111ARG TRAINING_RESULTS_REPO=https://github.com/mlcommons/training_results_v6.0.git0 B
112ARG TRAINING_RESULTS_COMMIT=eabf23a07b2a0c60a289ff871dc3a46fff0d04210 B
113RUN |31 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d848.1 KB
114RUN |31 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8256 B
115RUN |31 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8510 B
116RUN |31 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d832 B
117RUN |31 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d832 B
118ARG GIT_COMMIT_TAG=3793dafa186b5a126aebea6762080b64abf533190 B
119ARG DOCKERFILE_PATH=primus_training_the_rock/docker/Dockerfile0 B
120RUN |33 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260727 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d822.86 KB
121COPY primus_training_the_rock/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit5.91 KB
122CMD ["/usr/bin/bash"]0 B

Command

 ARG RELEASE