Sign inSign up

genesu100/utc_ci:30959889346

Index digest

sha256:4da1208a844e69a8cae26aa51c29f58fbba1141ce71fd314a714287b1971ba7f

OS/ARCH

linux/amd64

Compressed size

13.51 GB

Last pushed

about 1 month by genesu100

Type

Image

Manifest digest

sha256:8f450191586ef0dcc1586169981ff070021a2b6d04d418ed51f3ca8a494e8851

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/97 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 apt update242.81 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
15RUN /bin/sh -c pip install33.29 MB
16RUN /bin/sh -c apt update94.9 MB
17ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
18ENV HSA_NO_SCRATCH_RECLAIM=10 B
19ENV THE_ROCK_VERSION=10.1.0a202608030 B
20RUN /bin/sh -c python -m3.25 GB
21ARG PYTORCH_VERSION=2.12.0+rocm10.1.0a202608030 B
22RUN |1 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 /bin/sh -c73.57 MB
23RUN |1 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 /bin/sh -c981.98 MB
24RUN |1 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 /bin/sh -c3.25 GB
25ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
26ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
27ENV HIP_PLATFORM=amd0 B
28ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
29ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
30ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
31ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
32ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
33ENV 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
34ENV 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
35ENV 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
36ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
37ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
38ARG TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf80 B
39ENV NVTE_USE_CAST_TRANSPOSE_TRITON=10 B
40ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B
41ENV NVTE_CK_USES_BWD_V3=10 B
42ENV NVTE_CK_USES_FWD_V3=10 B
43ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B
44ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
45RUN |2 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 /bin/sh25.96 MB
46RUN |2 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 /bin/sh322.24 MB
47ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
48ARG FLUX_BRANCH=26.4_RC_Patch_torchcodec0 B
49RUN |4 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git105.15 MB
50ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B
51ARG DLRM_BRANCH=main0 B
52RUN |6 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git103.58 KB
53ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
54ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
55RUN |8 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git21.72 MB
56ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
57ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B
58RUN |10 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git3.61 MB
59RUN |10 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git207.42 MB
60ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
61ARG GROUPED_GEMM_BRANCH=rocm0 B
62RUN |12 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git112.58 KB
63ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B
64ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B
65ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B
66ENV MAMBA_FORCE_BUILD=TRUE0 B
67ENV HIP_ARCHITECTURES=gfx942,gfx9500 B
68RUN |14 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git9.16 MB
69ARG MAMBA_REPO=https://github.com/AndreasKaratzas/mamba.git0 B
70ARG MAMBA_BRANCH=enable-primus-hybrid-models0 B
71RUN |16 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git276.92 MB
72ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
73ARG PRIMUS_BRANCH=885d036e95c8691a7e2414010a454042594cca180 B
74ENV NVTE_FLASH_ATTN=00 B
75ENV NVTE_FUSED_ATTN=10 B
76RUN |18 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git1.2 GB
77ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
78ARG AITER_COMMIT=0f3c58e6edb6754940bcf9fd5f09ccb6f389f52e0 B
79RUN |20 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git226.95 MB
80ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
81ARG TURBO_COMMIT=9b5d3092efcbc087657b233d8e9ae662cee6ec6b0 B
82ENV HCC_AMDGPU_TARGET=gfx942,gfx9500 B
83RUN |22 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git852.45 MB
84RUN |22 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git73.28 MB
85RUN |22 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git13.32 MB
86RUN |22 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git64.34 MB
87WORKDIR /workspace/32 B
88RUN |22 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git423.94 MB
89ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B
90ARG FBGEMM_COMMIT=80bd3c077dc41b55cd16ed4dcad15cf7c1c1d76a0 B
91ENV BUILD_ROCM_VERSION=7.150 B
92RUN |24 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git872.2 KB
93RUN |24 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git1.7 GB
94RUN |24 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git301.03 KB
95ENV GPU_ARCHS=native0 B
96RUN |24 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git8 MB
97RUN |24 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git496 B
98RUN |24 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git13.8 MB
99ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
100RUN |25 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git444.57 KB
101RUN |25 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git228 B
102ARG UCX_VERSION=1.18.00 B
103RUN |26 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git86.43 MB
104ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
105ARG MPI_VERSION=4.1.60 B
106RUN |27 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git3.84 MB
107ENV 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
108ENV 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
109ARG TRAINING_RESULTS_REPO=https://github.com/mlcommons/training_results_v6.0.git0 B
110ARG TRAINING_RESULTS_COMMIT=eabf23a07b2a0c60a289ff871dc3a46fff0d04210 B
111RUN |29 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git48.09 KB
112RUN |29 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git257 B
113RUN |29 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git476 B
114RUN |29 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git32 B
115RUN |29 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git32 B
116ARG GIT_COMMIT_TAG=34fc1e7c7d945ba25092a30fbcc98ff59cc41be10 B
117ARG DOCKERFILE_PATH=primus_training_the_rock/docker/Dockerfile0 B
118RUN |31 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260803 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git20.05 KB
119COPY primus_training_the_rock/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit6.11 KB
120CMD ["/usr/bin/bash"]0 B

Command

 ARG RELEASE