Sign inSign up

genesu100/utc_ci:30070836133

Index digest

sha256:3ffbb330aed0bb93835ad1df79332aab82086ac6d62d4aa8b3670dc91e887aba

OS/ARCH

linux/amd64

Compressed size

13.35 GB

Last pushed

3 months by genesu100

Type

Image

Manifest digest

sha256:30e0b7875e23945588d50236ae5e47c8d8c49bcafa5360bef4a51193c5c508ac

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 28.36 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 update276.63 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.32 MB
15RUN /bin/sh -c apt update137.53 MB
16ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
17ENV HSA_NO_SCRATCH_RECLAIM=10 B
18ARG PYTORCH_VERSION=2.12.0+rocm7.15.0a202607240 B
19RUN |1 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 /bin/sh -c73.57 MB
20RUN |1 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 /bin/sh -c4.24 GB
21RUN |1 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 /bin/sh -c3.28 GB
22ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
23ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
24ENV HIP_PLATFORM=amd0 B
25ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
26ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
27ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
28ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
29ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
30ENV 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
31ENV 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
32ENV 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
33ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
34ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
35ARG FA_REPO=https://github.com/ROCm/flash-attention.git0 B
36ARG FA_BRANCH=6387433156558135a998d5568a9d74c1778666d80 B
37ENV GPU_ARCHS=gfx942;gfx9500 B
38RUN |3 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d873.02 MB
39ARG TE_VERSION=2.15.0.dev0+rocm7.15.0a20260721.6e541a10 B
40RUN |4 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d825.79 MB
41RUN |4 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8324.92 MB
42ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
43ARG FLUX_BRANCH=26.4_RC_Patch_torchcodec0 B
44RUN |6 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8105.1 MB
45ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B
46ARG DLRM_BRANCH=main0 B
47RUN |8 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8103.56 KB
48ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
49ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
50RUN |10 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d821.69 MB
51ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
52ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B
53RUN |12 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d83.61 MB
54RUN |12 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8207.18 MB
55ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
56ARG GROUPED_GEMM_BRANCH=rocm0 B
57RUN |14 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8112.55 KB
58ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B
59ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B
60ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B
61ENV MAMBA_FORCE_BUILD=TRUE0 B
62ENV HIP_ARCHITECTURES=gfx942,gfx9500 B
63RUN |16 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d89.16 MB
64ARG MAMBA_REPO=https://github.com/AndreasKaratzas/mamba.git0 B
65ARG MAMBA_BRANCH=enable-primus-hybrid-models0 B
66RUN |18 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8276.92 MB
67ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
68ARG PRIMUS_BRANCH=551bc75216db55f3b51f1103001ee26c0e56b08e0 B
69ENV NVTE_FLASH_ATTN=00 B
70ENV NVTE_FUSED_ATTN=10 B
71RUN |20 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d81.17 GB
72ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
73ARG AITER_COMMIT=b5e03ed191fca11ee423226537ef8d9435e432a60 B
74RUN |22 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8226.64 MB
75ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
76ARG TURBO_COMMIT=3c39ef259aa6d724c77c481e926466e7a167e9380 B
77ENV MAX_JOBS=1280 B
78ENV HCC_AMDGPU_TARGET=gfx942,gfx9500 B
79RUN |24 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8639.91 MB
80RUN |24 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d813.36 MB
81WORKDIR /workspace/32 B
82RUN |24 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8423.85 MB
83ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B
84ARG FBGEMM_COMMIT=80bd3c077dc41b55cd16ed4dcad15cf7c1c1d76a0 B
85ENV BUILD_ROCM_VERSION=7.140 B
86RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8975.23 KB
87RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d81.7 GB
88RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8377.41 KB
89ENV GPU_ARCHS=native0 B
90RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d88.08 MB
91RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8859.36 KB
92RUN |26 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d811.07 MB
93ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
94RUN |27 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d82.74 MB
95RUN |27 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8230 B
96ARG UCX_VERSION=1.18.00 B
97RUN |28 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d886.43 MB
98ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
99ARG MPI_VERSION=4.1.60 B
100RUN |29 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d83.84 MB
101ENV 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
102ENV 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
103ARG ROCSHMEM_REPO=https://github.com/ROCm/rocm-systems0 B
104ARG ROCSHMEM_BRANCH=64887dbf2e390262fa327badb490949a3eecdcd80 B
105ENV ROCSHMEM_HOME=/opt/rocshmem0 B
106ENV UCX_HOME=/workspace/ucx-1.18.0/install0 B
107ENV MPI_HOME=/opt/openmpi0 B
108RUN |31 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d818.23 MB
109ARG TRAINING_RESULTS_REPO=https://github.com/mlcommons/training_results_v6.0.git0 B
110ARG TRAINING_RESULTS_COMMIT=eabf23a07b2a0c60a289ff871dc3a46fff0d04210 B
111RUN |33 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d848.1 KB
112RUN |33 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8255 B
113RUN |33 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d8445 B
114RUN |33 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d832 B
115RUN |33 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d832 B
116ARG GIT_COMMIT_TAG=e38bdef2d961a1556ec6028a2731cd5e3109db180 B
117ARG DOCKERFILE_PATH=primus_training_the_rock/docker/Dockerfile0 B
118RUN |35 PYTORCH_VERSION=2.12.0+rocm7.15.0a20260724 FA_REPO=https://github.com/ROCm/flash-attention.git FA_BRANCH=6387433156558135a998d5568a9d74c1778666d822.93 KB
119COPY primus_training_the_rock/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit5.26 KB
120CMD ["/usr/bin/bash"]0 B

Command

 ARG RELEASE