Sign inSign up

genesu100/utc_ci:33437972515

Index digest

sha256:cfc1321cad6e10a76d965c0fc02af812e06d74ca91634ccda2306f114c1e4f1d

OS/ARCH

linux/amd64

Compressed size

13.28 GB

Last pushed

27 days by genesu100

Type

Image

Manifest digest

sha256:8ee6064c0772a288e86e01ffa51b8a2a76c7709b77d1efb14a531e042b521582

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 MLPERF_RUNTIME_SERIES=v26.50 B
12ENV MLPERF_ENABLE_FWD_ATTN_ASM=10 B
13ENV FMHA_HD64_ASM_CO=/opt/venv/lib/python3.12/site-packages/aiter_hd64_asm_fwd_d64_opt128.co0 B
14ENV FMHA_HD64_ASM_LOG=00 B
15ENV DEBIAN_FRONTEND=noninteractive0 B
16RUN /bin/sh -c apt update245.17 MB
17COPY /uv /usr/local/bin/uv # buildkit20.96 MB
18RUN /bin/sh -c uv venv9.65 KB
19ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
20ENV VIRTUAL_ENV=/opt/venv0 B
21COPY /dist/pip-*.whl /tmp/ # buildkit1.71 MB
22RUN /bin/sh -c uv pip29.08 MB
23RUN /bin/sh -c apt update96.4 MB
24ENV HSA_ENABLE_SCRATCH_ASYNC_RECLAIM=00 B
25ENV HSA_NO_SCRATCH_RECLAIM=10 B
26ENV THE_ROCK_VERSION=10.1.0a202608220 B
27RUN /bin/sh -c uv pip3.63 GB
28ARG PYTORCH_VERSION=2.12.0+rocm10.1.0a202608220 B
29RUN |1 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 /bin/sh -c52.18 MB
30RUN |1 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 /bin/sh -c947.17 MB
31RUN |1 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 /bin/sh -c3.63 GB
32ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
33ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
34ENV HIP_PLATFORM=amd0 B
35ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
36ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
37ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
38ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
39ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
40ENV 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
41ENV 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
42ENV 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
43ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
44ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
45RUN |1 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 /bin/sh -c24.51 KB
46ARG FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git0 B
47ARG FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba220 B
48ARG BWD_ATTN_ASM_REPO=https://github.com/mawad-amd/bwd-attn-asm.git0 B
49ARG BWD_ATTN_ASM_REF=9b9fb6444f3fee388617f62432c3faea740793770 B
50ARG BWD_ATTN_SOURCE_SYMBOL=_ZN5aiter43fmha_bwd_hd64_bf16_causal_a16_rtz_recompileE0 B
51ARG BWD_ATTN_SYMBOL=_ZN5aiter44fmha_bwd_hd64_bf16_causal_a16_rtne_recompileE0 B
52RUN |7 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2212.69 KB
53ARG TE_REPO=https://github.com/ROCm/TransformerEngine.git0 B
54ARG TE_VERSION=8af6efcd9a40afdf49277a1471778c5dbe4ca7040 B
55ARG BWD_ATTN_SLOT=bwd_hd64_bf16_causal_a16_rtne.co0 B
56ENV NVTE_USE_HIPBLASLT=10 B
57ENV NVTE_FRAMEWORK=pytorch0 B
58ENV NVTE_ROCM_ARCH=gfx942;gfx9500 B
59ENV NVTE_USE_CAST_TRANSPOSE_TRITON=10 B
60ENV NVTE_CK_IS_V3_ATOMIC_FP32=00 B
61ENV NVTE_CK_USES_BWD_V3=10 B
62ENV NVTE_CK_USES_FWD_V3=10 B
63ENV CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=20 B
64ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
65ENV NVTE_USE_ROCM=10 B
66ENV NVTE_FUSED_ATTN_AOTRITON=00 B
67ENV NVTE_FUSED_ATTN_CK=10 B
68ENV NVTE_CK_JIT=10 B
69ENV NVTE_CK_FUSED_ATTN_PATH=0 B
70ENV NVTE_AITER_SOURCE_DIR=/workspace/deps/te-aiter0 B
71ENV NVTE_BUILD_MAX_JOBS=1280 B
72ENV CU_NUM=3040 B
73ENV NVTE_SKIP_SUBMODULE_CHECKS_DURING_BUILD=10 B
74ENV CMAKE_BUILD_PARALLEL_LEVEL=1280 B
75RUN |10 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22482.28 MB
76RUN |10 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2264.36 MB
77ARG FLUX_REPO=https://github.com/ROCm/AMDiffusionBenchmark.git0 B
78ARG FLUX_BRANCH=26.4_RC_Patch_torchcodec0 B
79COPY primus_training_the_rock/docker/Diffusion_benchmark_patches.patch /workspace/Diffusion_benchmark_patches.patch # buildkit2.81 KB
80COPY primus_training_the_rock/docker/sdxl_patch.patch /workspace/sdxl_patch.patch # buildkit581 B
81RUN |12 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2285.65 MB
82ARG DLRM_REPO=https://github.com/AMD-AGI/DLRMBenchmark.git0 B
83ARG DLRM_BRANCH=main0 B
84RUN |14 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22103.56 KB
85ARG TORCHTUNE_REPO=https://github.com/pytorch/torchtune.git0 B
86ARG TORCHTUNE_BRANCH=b4c98ac2a37f0397d64c22579aed415ce7264db60 B
87RUN |16 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2217.11 MB
88ARG TORCHAO_REPO=https://github.com/pytorch/ao.git0 B
89ARG TORCHAO_BRANCH=e9c7bead90b840b280f97374308255957108ce470 B
90RUN |18 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba222.06 MB
91RUN |18 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22172.72 MB
92ARG GROUPED_GEMM_REPO=https://github.com/caaatch22/grouped_gemm.git0 B
93ARG GROUPED_GEMM_BRANCH=rocm0 B
94RUN |20 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22204.39 KB
95ARG CAUSAL_CONV1D_REPO=https://github.com/Dao-AILab/causal-conv1d0 B
96ARG CAUSAL_CONV1D_BRANCH=e940ead2fd962c56854455017541384909ca669f0 B
97ENV CAUSAL_CONV1D_FORCE_BUILD=TRUE0 B
98ENV MAMBA_FORCE_BUILD=TRUE0 B
99ENV HIP_ARCHITECTURES=gfx942,gfx9500 B
100RUN |22 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba229.14 MB
101ARG MAMBA_REPO=https://github.com/AndreasKaratzas/mamba.git0 B
102ARG MAMBA_BRANCH=enable-primus-hybrid-models0 B
103RUN |24 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2285.69 MB
104ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
105ARG PRIMUS_BRANCH=fd373fd349e7ad5a787c6c04efe3cd0c1bba0bcb0 B
106ENV NVTE_FLASH_ATTN=00 B
107ENV NVTE_FUSED_ATTN=10 B
108ENV PRIMUS_FLA_MLA_ATTN=10 B
109RUN |26 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba221.58 GB
110ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
111ARG AITER_COMMIT=0f3c58e6edb6754940bcf9fd5f09ccb6f389f52e0 B
112RUN |28 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22124.5 MB
113ARG TURBO_REPO=https://github.com/AMD-AGI/Primus-Turbo.git0 B
114ARG TURBO_COMMIT=a6a16cdce46bd2235a248b41f501fb363c215b9b0 B
115ENV HCC_AMDGPU_TARGET=gfx942,gfx9500 B
116ENV PRIMUS_TURBO_FRAMEWORK=PYTORCH0 B
117RUN |30 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22139.5 MB
118RUN |30 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2212.57 MB
119RUN |30 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2254.82 MB
120RUN |30 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2210 MB
121WORKDIR /workspace/32 B
122RUN |30 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba222.3 MB
123ARG FBGEMM_REPO=https://github.com/pytorch/FBGEMM.git0 B
124ARG FBGEMM_COMMIT=80bd3c077dc41b55cd16ed4dcad15cf7c1c1d76a0 B
125ENV BUILD_ROCM_VERSION=7.150 B
126RUN |32 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22873.65 KB
127RUN |32 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba221.7 GB
128RUN |32 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22302.41 KB
129ENV GPU_ARCHS=native0 B
130RUN |32 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba228 MB
131RUN |32 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22857 B
132RUN |32 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2213.74 MB
133ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
134RUN |33 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22445.93 KB
135RUN |33 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22230 B
136ARG UCX_VERSION=1.18.00 B
137RUN |34 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2286.43 MB
138ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
139ARG MPI_VERSION=4.1.60 B
140RUN |35 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba223.84 MB
141ENV 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
142ENV 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
143ARG MLPERF_COMMENT_REPO=https://github.com/AMD-AGI/mlperf-common.git0 B
144ARG MLPERF_COMMENT_COMMIT=56d8ca70f60f866312add84adfea66f5447a4c860 B
145RUN |37 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2224.98 KB
146COPY /wandb-core /opt/venv/lib/python3.12/site-packages/wandb/bin/wandb-core # buildkit15.15 MB
147RUN |37 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22258 B
148RUN |37 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22546 B
149RUN |37 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2232 B
150RUN |37 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2232 B
151RUN |37 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba22401 B
152ARG GIT_COMMIT_TAG=6acb2d972113b75f70fc99b9d06a4116e336e6b70 B
153ARG DOCKERFILE_PATH=primus_training_the_rock/docker/Dockerfile0 B
154RUN |39 PYTORCH_VERSION=2.12.0+rocm10.1.0a20260822 FWD_ATTN_ASM_REPO=https://github.com/mawad-amd/fwd-attn-asm.git FWD_ATTN_ASM_REF=53d3dadc3f3b0ac35ae536f2d1d7864a3e07ba2221.14 KB
155COPY primus_training_the_rock/docker/Dockerfile /workspace/.manifest/Dockerfile # buildkit10.29 KB
156CMD ["/usr/bin/bash"]0 B

Command

 ARG RELEASE