Sign inSign up

rocm/primus:v25.9_gfx950

Manifest digest

sha256:1a198be32f49efd66d0ff82066b44bd99b3e6b04c8e0e9b36b2c481e13bff7b6

OS/ARCH

linux/amd64

Compressed size

23.4 GB

Last pushed

10 months by rocmdev

Type

Image

Manifest digest

sha256:1a198be32f49efd66d0ff82066b44bd99b3e6b04c8e0e9b36b2c481e13bff7b6

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 / 29.04 MB
6CMD ["/bin/bash"]0 B
7ARG PYTHON_VERSION0 B
8ARG PYTORCH_VERSION0 B
9ARG ROCM_VERSION0 B
10ARG AMDGPU_VERSION0 B
11ENV DEBIAN_FRONTEND=noninteractive0 B
12RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.0279.22 KB
13RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.0110.13 MB
14ENV PATH=/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
15ENV LD_LIBRARY_PATH=/opt/rocm/lib0 B
16RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.096.98 MB
17RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.07.91 MB
18ENV PATH=/opt/venv/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
19RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.016.65 KB
20RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.0390.53 KB
21RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.05.77 GB
22RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.02.72 GB
23RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.6.0 ROCM_VERSION=7.0104.75 KB
24ENV ROCM_PATH=/opt/rocm-7.0.00 B
25WORKDIR /workspace/99 B
26RUN /bin/sh -c apt --fix-broken393 B
27RUN /bin/sh -c pip install690.86 KB
28ENV MAX_JOBS=1280 B
29ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
30ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
31COPY /*.whl build/ # buildkit1.62 GB
32RUN /bin/sh -c pip install1.75 GB
33RUN /bin/sh -c rm build/*327 B
34RUN /bin/sh -c pip uninstall51.85 MB
35RUN /bin/sh -c pip uninstall891.5 KB
36RUN /bin/sh -c apt install3.26 MB
37RUN /bin/sh -c cd /opt/rocm/share/amd_smi1.52 MB
38ARG HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git0 B
39ARG HIPBLASLT_BRANCH=634088c98d0 B
40RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d /bin/sh183 B
41RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d /bin/sh2.43 GB
42RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d /bin/sh182 B
43RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d /bin/sh1.1 GB
44ARG FA_REPO=https://github.com/ROCm/flash-attention.git0 B
45ARG FA_BRANCH=6387433156558135a998d5568a9d74c1778666d80 B
46ENV GPU_ARCHS=gfx942;gfx9500 B
47RUN |4 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git93.15 MB
48ARG TE_REPO=https://github.com/ROCm/TransformerEngine.git0 B
49ARG TE_BRANCH=c3bcaab1311d3d961880733eeab87ea7391adb990 B
50ENV NVTE_USE_HIPBLASLT=10 B
51ENV NVTE_FRAMEWORK=pytorch0 B
52ENV NVTE_ROCM_ARCH=gfx942;gfx9500 B
53ENV NVTE_USE_CAST_TRANSPOSE_TRITON=00 B
54ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B
55ENV NVTE_CK_USES_BWD_V3=10 B
56ENV NVTE_CK_USES_FWD_V3=10 B
57ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B
58ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
59ENV NVTE_USE_ROCM=10 B
60ENV GPU_ARCHS=gfx9500 B
61ENV CU_NUM=3040 B
62RUN |6 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git4.34 GB
63ARG AITER_REPO=https://github.com/rocm/aiter.git0 B
64ARG AITER_BRANCH=main0 B
65ENV AITER_ASM_DIR=/workspace/TransformerEngine/3rdparty/aiter/hsa/gfx950/0 B
66RUN |8 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git171.99 MB
67ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
68ARG FLUX_BRANCH=Flux_patch0 B
69RUN |10 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git139.01 MB
70ARG TORCHTITAN_REPO=https://github.com/AMD-AIG-AIMA/torchtitan.git0 B
71ARG TORCHTITAN_BRANCH=dev/aiter_attn_random_input0 B
72RUN |12 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git122.89 MB
73ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
74ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
75RUN |14 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git124.62 MB
76ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
77ARG TORCHAO_BRANCH=a9ffa508ef191eb69119658827cdda90dd918dca0 B
78RUN |16 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git2.93 MB
79RUN |16 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git37.4 MB
80RUN |16 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git957.56 KB
81RUN |16 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git1.38 KB
82RUN |16 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git3.62 MB
83RUN |16 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git129.67 MB
84RUN |16 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git1.45 MB
85RUN |16 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git8.19 MB
86ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
87ARG GROUPED_GEMM_BRANCH=rocm0 B
88RUN |18 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git99.68 KB
89ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B
90ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B
91ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B
92ENV MAMBA_FORCE_BUILD=TRUE0 B
93ENV HIP_ARCHITECTURES=gfx942,gfx9500 B
94RUN |20 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git16.28 MB
95ARG ROCSHMEM_REPO=https://github.com/ROCm/rocSHMEM.git0 B
96ARG ROCSHMEM_BRANCH=e95360961dfa5575a49fe79abdf41ab9e4aeb5430 B
97RUN |22 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git99.86 MB
98ARG MEGATRON_REPO=https://github.com/ROCm/Megatron-LM.git0 B
99ARG MEGATRON_BRANCH=yaof/dev/mi355_optimization0 B
100RUN |24 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git86.21 MB
101ARG PRIMUS_REPO=https://github.com/AMD-AIG-AIMA/Primus.git0 B
102ARG PRIMUS_BRANCH=main0 B
103RUN |26 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git150.19 MB
104ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
105ARG TURBO_COMMIT=main0 B
106ENV MAX_JOBS=1280 B
107ENV GPU_ARCHS=gfx9500 B
108RUN |28 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git66.53 MB
109RUN |28 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=634088c98d FA_REPO=https://github.com/ROCm/flash-attention.git321 B
110CMD ["/usr/bin/bash"]0 B
111WORKDIR /workspace/32 B
112ARG MEGATRON_REPO=https://github.com/ROCm/Megatron-LM.git0 B
113ARG MEGATRON_BRANCH=yaof/dev/mi355_optimization0 B
114RUN |2 MEGATRON_REPO=https://github.com/ROCm/Megatron-LM.git MEGATRON_BRANCH=yaof/dev/mi355_optimization /bin/sh35.77 MB
115ARG GPU_ARCHS=gfx9500 B
116ENV GPU_ARCHS=gfx9500 B
117RUN |3 MEGATRON_REPO=https://github.com/ROCm/Megatron-LM.git MEGATRON_BRANCH=yaof/dev/mi355_optimization GPU_ARCHS=gfx950201.91 MB
118RUN |3 MEGATRON_REPO=https://github.com/ROCm/Megatron-LM.git MEGATRON_BRANCH=yaof/dev/mi355_optimization GPU_ARCHS=gfx95030.83 MB
119RUN |3 MEGATRON_REPO=https://github.com/ROCm/Megatron-LM.git MEGATRON_BRANCH=yaof/dev/mi355_optimization GPU_ARCHS=gfx95032 B
120RUN /bin/sh -c rm -rf135 B
121ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
122ARG PRIMUS_BRANCH=5faa9ce4785eaf2a05f7ead123369f78ce7749520 B
123RUN |2 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=5faa9ce4785eaf2a05f7ead123369f78ce774952 /bin/sh154.8 MB
124ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B
125ARG TRITON_BRANCH=1cc4a7c4aac921b7fa4658e162058632b7faed550 B
126RUN |4 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=5faa9ce4785eaf2a05f7ead123369f78ce774952 TRITON_REPO=https://github.com/ROCm/triton.git1.74 GB
127RUN /bin/sh -c cd Primus230.42 KB
128RUN /bin/sh -c rm /workspace/Primus/.git/packed-refs313 B

Command

 ARG RELEASE