Sign inSign up

rocm/primus:v26.7

Manifest digest

sha256:68b7eb7d4db99ecfa18bd7972e5b6d8b78e56219f0a144a8ef5e9a993e57f195

OS/ARCH

linux/amd64

Compressed size

13.35 GB

Last pushed

8 days by rocm

Type

Image

Manifest digest

sha256:68b7eb7d4db99ecfa18bd7972e5b6d8b78e56219f0a144a8ef5e9a993e57f195

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 GPU_ARCHS=gfx942;gfx9500 B
11ENV DEBIAN_FRONTEND=noninteractive0 B
12RUN /bin/sh -c echo "cache-bust246.51 MB
13RUN /bin/sh -c python -m4 MB
14ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
15COPY /dist/pip-*.whl /tmp/ # buildkit1.71 MB
16RUN /bin/sh -c pip install32.89 MB
17RUN /bin/sh -c apt update96.3 MB
18ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
19ENV HSA_NO_SCRATCH_RECLAIM=10 B
20ENV THE_ROCK_VERSION=10.0.00 B
21RUN /bin/sh -c python -m3.26 GB
22ARG PYTORCH_VERSION=2.12.0+rocm10.0.00 B
23RUN |1 PYTORCH_VERSION=2.12.0+rocm10.0.0 /bin/sh -c99.34 MB
24RUN |1 PYTORCH_VERSION=2.12.0+rocm10.0.0 /bin/sh -c955.88 MB
25RUN |1 PYTORCH_VERSION=2.12.0+rocm10.0.0 /bin/sh -c3.25 GB
26ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
27ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
28ENV HIP_PLATFORM=amd0 B
29ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
30ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
31ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
32ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
33ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
34ENV 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
35ENV 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
36ENV 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
37ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
38ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
39RUN |1 PYTORCH_VERSION=2.12.0+rocm10.0.0 /bin/sh -c23.87 KB
40ARG TE_VERSION=2.17.0+rocm10.0.00 B
41ENV NVTE_USE_CAST_TRANSPOSE_TRITON=10 B
42ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B
43ENV NVTE_CK_USES_BWD_V3=10 B
44ENV NVTE_CK_USES_FWD_V3=10 B
45ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B
46ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
47RUN |2 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 /bin/sh380.46 MB
48RUN |2 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 /bin/sh64.15 MB
49ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
50ARG FLUX_BRANCH=26.4_RC_Patch_torchcodec0 B
51COPY primus_training_the_rock/docker/Diffusion_benchmark_patches.patch /workspace/Diffusion_benchmark_patches.patch # buildkit2.81 KB
52COPY primus_training_the_rock/docker/sdxl_patch.patch /workspace/sdxl_patch.patch # buildkit581 B
53RUN |4 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git105.63 MB
54ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B
55ARG DLRM_BRANCH=main0 B
56RUN |6 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git103.55 KB
57ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
58ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
59RUN |8 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git21.89 MB
60ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
61ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B
62RUN |10 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git3.61 MB
63RUN |10 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git207.23 MB
64ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
65ARG GROUPED_GEMM_BRANCH=rocm0 B
66RUN |12 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git112.58 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 |14 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git9.16 MB
73ARG MAMBA_REPO=https://github.com/AndreasKaratzas/mamba.git0 B
74ARG MAMBA_BRANCH=enable-primus-hybrid-models0 B
75RUN |16 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git86.14 MB
76ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
77ARG PRIMUS_BRANCH=2631e68dd8b658ab1f991cbc671d538205a51fee0 B
78ENV NVTE_FLASH_ATTN=00 B
79ENV NVTE_FUSED_ATTN=10 B
80ENV PRIMUS_FLA_MLA_ATTN=10 B
81RUN |18 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git1.63 GB
82ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
83ARG AITER_COMMIT=0f3c58e6edb6754940bcf9fd5f09ccb6f389f52e0 B
84RUN |20 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git226.86 MB
85ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
86ARG TURBO_COMMIT=6d5ff979eb019fbbcd91790ac812024cca05a8820 B
87ENV HCC_AMDGPU_TARGET=gfx942,gfx9500 B
88RUN |22 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git783.24 MB
89RUN |22 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git13.32 MB
90RUN |22 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git64.68 MB
91RUN |22 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git15.23 MB
92WORKDIR /workspace/32 B
93RUN |22 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git6.25 MB
94ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B
95ARG FBGEMM_COMMIT=80bd3c077dc41b55cd16ed4dcad15cf7c1c1d76a0 B
96ENV BUILD_ROCM_VERSION=7.150 B
97RUN |24 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git876.09 KB
98RUN |24 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git1.72 GB
99RUN |24 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git305.11 KB
100ENV GPU_ARCHS=native0 B
101RUN |24 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git8.01 MB
102RUN |24 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git32 B
103RUN |24 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git13.96 MB
104ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
105RUN |25 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git449.09 KB
106RUN |25 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git230 B
107ARG UCX_VERSION=1.18.00 B
108RUN |26 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git86.43 MB
109ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
110ARG MPI_VERSION=4.1.60 B
111RUN |27 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git3.84 MB
112ENV 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
113ENV 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
114ARG MLPERF_COMMENT_REPO=https://github.com/AMD-AGI/mlperf-common.git0 B
115ARG MLPERF_COMMENT_COMMIT=56d8ca70f60f866312add84adfea66f5447a4c860 B
116RUN |29 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git57.33 KB
117COPY /wandb-core /opt/venv/lib/python3.12/site-packages/wandb/bin/wandb-core # buildkit15.16 MB
118RUN |29 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git258 B
119RUN |29 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git545 B
120RUN |29 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git32 B
121RUN |29 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git32 B
122RUN |29 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git32 B
123ARG GIT_COMMIT_TAG=db0de753f37d6a6b782bc836b2be54d8ae3622580 B
124ARG DOCKERFILE_PATH=primus_training_the_rock/docker/Dockerfile0 B
125RUN |31 PYTORCH_VERSION=2.12.0+rocm10.0.0 TE_VERSION=2.17.0+rocm10.0.0 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git20.01 KB
126COPY primus_training_the_rock/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit9.29 KB
127CMD ["/usr/bin/bash"]0 B

Command

 ARG RELEASE