Sign inSign up

rocm/primus:v26.1

Manifest digest

sha256:4fc8808bdb14117c6af7f38d79c809056e6fdbfd530c1fabbb61d097ddaf820d

OS/ARCH

linux/amd64

Compressed size

48.48 GB

Last pushed

7 months by rocmdev

Type

Image

Manifest digest

sha256:4fc8808bdb14117c6af7f38d79c809056e6fdbfd530c1fabbb61d097ddaf820d

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 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.8.0 ROCM_VERSION=7.1279.21 KB
13RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.1111.82 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.8.0 ROCM_VERSION=7.196.99 MB
17RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.17.96 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.8.0 ROCM_VERSION=7.116.68 KB
20RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.1390.62 KB
21RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.16.45 GB
22RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.13.69 GB
23RUN |4 PYTHON_VERSION=3.10 PYTORCH_VERSION=2.8.0 ROCM_VERSION=7.1120.57 KB
24ENV ROCM_PATH=/opt/rocm-7.1.00 B
25WORKDIR /workspace/99 B
26RUN /bin/sh -c pip install690.92 KB
27ENV MAX_JOBS=1280 B
28ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
29ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
30RUN /bin/sh -c pip install10.3 GB
31RUN /bin/sh -c pip uninstall51.86 MB
32RUN /bin/sh -c pip uninstall891.5 KB
33RUN /bin/sh -c pip uninstall221.58 KB
34RUN /bin/sh -c apt install54.79 MB
35RUN /bin/sh -c cd /opt/rocm/share/amd_smi2.07 MB
36ARG LLVM_REPO=https://github.com/ROCm/llvm-project.git0 B
37ARG LLVM_BRANCH=release/rocm-rel-7.10 B
38RUN |2 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 /bin/sh1.13 GB
39ARG HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git0 B
40ARG HIPBLASLT_BRANCH=09ab7153e2b569dcc37b8c483fdc9f52136679a40 B
41RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git64.12 MB
42RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git175 B
43RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git4.77 GB
44RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git185 B
45RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git172 B
46RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git196.2 KB
47RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git196.2 KB
48RUN |4 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git172 B
49ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B
50ARG TRITON_BRANCH=1cc4a7c4aac921b7fa4658e162058632b7faed550 B
51RUN |6 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.74 GB
52ARG VISION_REPO=https://github.com/pytorch/vision0 B
53ARG VISION_BRANCH=acccf86477759b2d3500f1ae1be065f7b1e409ec0 B
54RUN |8 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.84 MB
55ARG FA_REPO=https://github.com/ROCm/flash-attention.git0 B
56ARG FA_BRANCH=6387433156558135a998d5568a9d74c1778666d80 B
57ENV GPU_ARCHS=gfx942;gfx9500 B
58RUN |10 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git90.3 MB
59ARG AITER_REPO=https://github.com/rocm/aiter.git0 B
60ARG AITER_BRANCH=7a41cca67187bd5f77c337765a1a289337901cef0 B
61RUN |12 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.49 GB
62ARG TE_REPO=https://github.com/ROCm/TransformerEngine.git0 B
63ARG TE_BRANCH=32e2d1d4d4af5976970dc3ca21bb17c97821a2390 B
64ENV NVTE_USE_HIPBLASLT=10 B
65ENV NVTE_FRAMEWORK=pytorch0 B
66ENV NVTE_ROCM_ARCH=gfx942;gfx9500 B
67ENV NVTE_USE_CAST_TRANSPOSE_TRITON=10 B
68ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B
69ENV NVTE_CK_USES_BWD_V3=10 B
70ENV NVTE_CK_USES_FWD_V3=10 B
71ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B
72ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
73ENV NVTE_USE_ROCM=10 B
74ENV NVTE_CK_FUSED_ATTN_PATH=/workspace/aiter/op_tests/cpp/mha0 B
75RUN |14 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git3.74 GB
76ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
77ARG FLUX_BRANCH=Flux_patch0 B
78RUN |16 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git176.32 MB
79ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B
80ARG DLRM_BRANCH=main0 B
81RUN |18 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git92.53 KB
82ARG TORCHTITAN_REPO=https://github.com/AMD-AIG-AIMA/torchtitan.git0 B
83ARG TORCHTITAN_BRANCH=dev/aiter_attn_random_input0 B
84RUN |20 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git124.75 MB
85ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
86ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
87RUN |22 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git117.66 MB
88ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
89ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B
90RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git3.06 MB
91RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git118.13 MB
92RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git970.76 KB
93RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.37 KB
94RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git3.73 MB
95RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git129.89 MB
96RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.45 MB
97RUN |24 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git8.19 MB
98ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
99ARG GROUPED_GEMM_BRANCH=rocm0 B
100RUN |26 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git108.48 KB
101ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B
102ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B
103ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B
104ENV MAMBA_FORCE_BUILD=TRUE0 B
105ENV HIP_ARCHITECTURES=gfx942,gfx9500 B
106RUN |28 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git16.63 MB
107ARG ROCSHMEM_REPO=https://github.com/ROCm/rocSHMEM.git0 B
108ARG ROCSHMEM_BRANCH=e95360961dfa5575a49fe79abdf41ab9e4aeb5430 B
109RUN |30 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git99.86 MB
110ARG MEGATRON_REPO=https://github.com/ROCm/Megatron-LM.git0 B
111ARG MEGATRON_BRANCH=feature/fsdp_optimization0 B
112RUN |32 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git121.69 MB
113ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
114ARG PRIMUS_BRANCH=700fa2e17f378cf897f62fe791a9c23aa78decb90 B
115RUN |34 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git409.48 MB
116ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
117ARG TURBO_COMMIT=de6aa038a141e7b8b5064131196383b733fb61150 B
118ENV MAX_JOBS=1280 B
119RUN |36 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git238.62 MB
120RUN |36 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git30.84 MB
121WORKDIR /workspace/32 B
122RUN |36 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git2.18 MB
123RUN |36 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git3.46 MB
124ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B
125ARG FBGEMM_COMMIT=main0 B
126ENV BUILD_ROCM_VERSION=7.10 B
127ENV FBGEMM_TBE_ROCM_HIP_BACKWARD_KERNEL=10 B
128ENV ROCM_VERSION=710000 B
129ENV HIPBLAS_V2=10 B
130RUN |38 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git1.18 GB
131ENV GPU_ARCHS=native0 B
132RUN |38 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git328 B
133RUN |38 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git308 B
134RUN |38 LLVM_REPO=https://github.com/ROCm/llvm-project.git LLVM_BRANCH=release/rocm-rel-7.1 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git126 B
135CMD ["/usr/bin/bash"]0 B
136ENV GPU_ARCHS=gfx942;gfx9500 B
137ARG TORCHAUDIO_REPO=https://github.com/pytorch/audio0 B
138ARG TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd0 B
139RUN |2 TORCHAUDIO_REPO=https://github.com/pytorch/audio TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd /bin/sh780.87 MB
140ARG APEX_REPO=https://github.com/rocm/apex.git0 B
141ARG APEX_BRANCH=release/1.9.00 B
142RUN |4 TORCHAUDIO_REPO=https://github.com/pytorch/audio TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd APEX_REPO=https://github.com/rocm/apex.git83.96 MB
143ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
144ARG PRIMUS_BRANCH=release/v25.100 B
145RUN |6 TORCHAUDIO_REPO=https://github.com/pytorch/audio TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd APEX_REPO=https://github.com/rocm/apex.git135 B
146RUN |6 TORCHAUDIO_REPO=https://github.com/pytorch/audio TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd APEX_REPO=https://github.com/rocm/apex.git571.49 MB
147ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
148ARG TURBO_COMMIT=docker-release-v25.100 B
149RUN |8 TORCHAUDIO_REPO=https://github.com/pytorch/audio TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd APEX_REPO=https://github.com/rocm/apex.git141 B
150RUN |8 TORCHAUDIO_REPO=https://github.com/pytorch/audio TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd APEX_REPO=https://github.com/rocm/apex.git136.3 MB
151ENV GPU_ARCHS=native0 B
152RUN |8 TORCHAUDIO_REPO=https://github.com/pytorch/audio TORCHAUDIO_COMMIT=8ad2aa5d354d1bf432339113860185d5a5d1abbd APEX_REPO=https://github.com/rocm/apex.git126 B
153RUN /bin/sh -c rm -f342 B
154ENV GPU_ARCHS=gfx942;gfx9500 B
155ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
156ARG PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e70 B
157RUN |2 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e7 /bin/sh135 B
158RUN |2 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e7 /bin/sh553.86 MB
159ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
160ARG TURBO_COMMIT=935a6314b1e6f57894c198a0b8def3053a8e903f0 B
161RUN |4 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e7 TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git141 B
162RUN |4 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e7 TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git151.07 MB
163ENV GPU_ARCHS=native0 B
164RUN |4 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e7 TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git254 B
165RUN |4 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e7 TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git384 B
166RUN |4 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e7 TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git124 B
167ARG GIT_COMMIT_TAG=b072c4a333bab864609ca0c388244917672bf9b60 B
168RUN |5 PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git PRIMUS_BRANCH=4bccca9052548db927f1f7dcfff25f0cd6c5c4e7 TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git186 B
169CMD ["/usr/bin/bash"]0 B
170ENV GPU_ARCHS=gfx942;gfx9500 B
171ARG HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git0 B
172ARG HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe0 B
173RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe /bin/sh172 B
174RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe /bin/sh4.71 GB
175RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe /bin/sh172 B
176RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe /bin/sh196.22 KB
177RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe /bin/sh196.21 KB
178RUN |2 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe /bin/sh172 B
179ARG TE_REPO=https://github.com/ROCm/TransformerEngine.git0 B
180ARG TE_BRANCH=f141f34bff6cd775dd113ee5a96f66c9d0a44fc80 B
181RUN |4 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git151 B
182RUN |4 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git3.65 GB
183ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
184ARG PRIMUS_BRANCH=9c529cd4a934a68a880ede036c3e97b792e381670 B
185RUN |6 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git135 B
186RUN |6 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git723.4 MB
187ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
188ARG TURBO_COMMIT=5233748e9c5c5795a6484ab31ece47c442d29ec20 B
189ENV HCC_AMDGPU_TARGET=gfx942,gfx9500 B
190RUN |8 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git145 B
191RUN |8 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git195.4 MB
192ENV GPU_ARCHS=native0 B
193RUN |8 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git256 B
194RUN |8 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git415 B
195RUN |8 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git126 B
196RUN |8 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git451 B
197ARG GIT_COMMIT_TAG=3fc3cfc3fd3cc702446169ca196332bd1a97a8d00 B
198RUN |9 HIPBLASLT_REPO=https://github.com/ROCm/rocm-libraries.git HIPBLASLT_BRANCH=34459f66eaf5ba4b8235adce2ad24926cef8bcbe TE_REPO=https://github.com/ROCm/TransformerEngine.git189 B
199CMD ["/usr/bin/bash"]0 B
200RUN /bin/sh -c apt update28.15 MB
201RUN /bin/sh -c add-apt-repository -y1.06 MB
202RUN /bin/sh -c echo 'Acquire::AllowInsecureRepositories232 B
203RUN /bin/sh -c wget https://github.com/openucx/ucx/releases/download/v1.18.0/ucx-1.18.0.tar.gz90.15 MB
204ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
205RUN /bin/sh -c wget https://download.open-mpi.org/release/open-mpi/v4.1/openmpi-4.1.6.tar.gz59.81 MB
206ENV OMPI_INSTALL_DIR=/workspace/ompi-4.1.6/install0 B
207RUN /bin/sh -c git clone222.17 MB
208ENV RCCL_HOME=/workspace/rccl0 B
209ENV MPI_PATH=/ompi-4.1.6/install0 B
210RUN /bin/sh -c apt install32.94 MB
211RUN /bin/sh -c git clone539.75 KB

Command

 ARG RELEASE