Sign inSign up

rocm/composable_kernel:ck_aiter

Manifest digest

sha256:4448c7e94d0b40a1f7019ae1150b04cdd1c81167c6d4a920410b6085d70a3d63

OS/ARCH

linux/amd64

Compressed size

25.68 GB

Last pushed

10 months by illsilin

Type

Image

Manifest digest

sha256:4448c7e94d0b40a1f7019ae1150b04cdd1c81167c6d4a920410b6085d70a3d63

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=24.040 B
5ADD file ... in / 28.34 MB
6CMD ["/bin/bash"]0 B
7ARG UBUNTU_VERSION0 B
8ENV DEBIAN_FRONTEND=noninteractive0 B
9ARG PYTORCH_ROCM_ARCH0 B
10ENV PYTORCH_ROCM_ARCH=gfx908;gfx90a;gfx1030;gfx1100;gfx1101;gfx942;gfx1200;gfx12010 B
11COPY file:b2a090cb02b5615fc2d1036ea77428cbb70253c6a8b8614f78fab85c2ba2dfb5 in install_base.sh 1.84 KB
12|1 UBUNTU_VERSION=24.04 /bin/sh -c bash320.15 MB
13ARG LLVMDEV0 B
14ARG CLANG_VERSION0 B
15COPY file:579317ceda1019f6cd72be303e1ec7c4c2486dbb81a333e2747d779070ed6794 in install_clang.sh 863 B
16|3 CLANG_VERSION= LLVMDEV= UBUNTU_VERSION=24.04 /bin/sh106 B
17COPY file:9f0cc865cc521cff044d880f1cb248b7f8e1ed60e8871fd662be075c2c6381b0 in install_user.sh 703 B
18|3 CLANG_VERSION= LLVMDEV= UBUNTU_VERSION=24.04 /bin/sh1.72 KB
19ARG ANACONDA_PYTHON_VERSION0 B
20ARG CONDA_CMAKE0 B
21ARG BUILD_ENVIRONMENT0 B
22ENV ANACONDA_PYTHON_VERSION=3.120 B
23ENV PATH=/opt/conda/envs/py_3.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
24COPY file:5bb2bf9f600c791c8add13c6b7fa012e87171d16f0ff9cc6e36b5fa85ef534e8 in /opt/conda/requirements-ci.txt 3.68 KB
25COPY file:4f2030250f761e6ac8f87880f43562c7195f8c965a62221ef19c1546241d8439 in install_conda.sh 1.91 KB
26COPY file:9e6db7f5d2e2c5e4a161d58370737ca06f20ca3a566e47689fc3c7bfeb0193a2 in common_utils.sh 724 B
27|5 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes LLVMDEV=2.71 GB
28ARG GCC_VERSION0 B
29COPY file:f0e99070d8114620f665268ab11f181c0153236fb98685afc2502a7f1e978b2c in install_gcc.sh 380 B
30|6 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION=104 B
31ARG PROTOBUF0 B
32COPY file:638eb6acc2fc2ff495707449113f05ffc78b46b8a4bb23639e30d89724372a1f in install_protobuf.sh 502 B
33|7 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION=92.5 MB
34|7 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes GCC_VERSION=109 B
35ENV INSTALLED_PROTOBUF=yes0 B
36ARG DB0 B
37COPY file:e2f252dfb0c5021405f1818c90ddc6a8fd573505ed7427bfbdd51167d3e5dda7 in install_db.sh 488 B
38|8 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes347 B
39|8 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes103 B
40ENV INSTALLED_DB=yes0 B
41ARG VISION0 B
42COPY multi:ecc701bddaf12d424d5629cf1acf0828dd4080da63bc673d1ddf7460edfab7e2 in ./ 1.84 KB
43|9 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes309.03 MB
44|9 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes163 B
45ENV INSTALLED_VISION=yes0 B
46ARG ROCM_VERSION0 B
47COPY file:97b22d5b61371bd4f097b3f607f7d0b6ded9cb3f757b0dc23b7c2375782f4ef6 in install_rocm.sh 1.65 KB
48|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes6.68 GB
49|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes105 B
50COPY file:dc007ec718680ca3fa2f00ec07dcee0d2f25ef694073fdb79ac48f76140b21d5 in install_rocm_magma.sh 1.09 KB
51|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes628.76 MB
52|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes111 B
53ADD file ... in install_miopen.sh 1.52 KB
54|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes107 B
55ENV ROCM_PATH=/opt/rocm0 B
56ENV PATH=/opt/rocm/bin:/opt/conda/envs/py_3.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
57ENV PATH=/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
58ENV PATH=/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
59ENV PATH=/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
60ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
61ENV MAGMA_HOME=/opt/rocm/magma0 B
62ENV LANG=C.UTF-80 B
63ENV LC_ALL=C.UTF-80 B
64COPY file:78e4a3281ca763191f3b77613d800ded4b7b00b181cdeec0200f96c67a3613ba in install_amdsmi.sh 166 B
65|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes2.8 MB
66|10 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CONDA_CMAKE=yes DB=yes107 B
67ARG CMAKE_VERSION0 B
68COPY file:5201a03ab45964625e825e856290df6dfb1826d702a824e47020a4d65c575468 in install_cmake.sh 566 B
69|11 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes41.21 MB
70|11 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes106 B
71ARG NINJA_VERSION0 B
72COPY file:f562ac7c4f4bdd63c08a52ddfc7e1180d5ee49530c1a18f726242695cfae1868 in install_ninja.sh 296 B
73|12 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes86.23 KB
74|12 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes104 B
75ARG TRITON0 B
76COPY file:4278720cb36f67567d70b5828bbe9eb47e3b03c9ae7a67fa254f9478559cd993 in install_triton.sh 1.37 KB
77COPY file:9e6db7f5d2e2c5e4a161d58370737ca06f20ca3a566e47689fc3c7bfeb0193a2 in common_utils.sh 724 B
78COPY file:77ff88eac004d39867271f88970329615a9fc3b28797c7bf4cc1caf5ede2c2f0 in triton.txt 138 B
79COPY file:317afa136ef0c0a5d80d6702afbfd60a0d37bb2987570d2514318ccc8051e223 in triton_version.txt 120 B
80|13 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes4.06 GB
81|13 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes172 B
82COPY file:62caab7ea7f18fb87560e07354fbefedb8b0eab2c4e3ed0bfa080fb1a2394d69 in install_cache.sh 1.89 KB
83ENV PATH=/opt/cache/bin:/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
84|13 BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.12 CLANG_VERSION= CMAKE_VERSION=3.18.5 CONDA_CMAKE=yes15.24 MB
85ARG BUILD_ENVIRONMENT0 B
86ENV BUILD_ENVIRONMENT=pytorch-linux-noble-rocm6.4.1-py3.120 B
87COPY dir:4ccc9e9d18635493a3825294a70103eaa912bdc81d80351d4dcd16f0a2e3d2b1 in /opt/llvm 51.64 MB
88USER jenkins0 B
89CMD ["bash"]0 B
90ARG FRAME_NAME0 B
91ARG OS_CHOICE0 B
92ARG PY_VER0 B
93ARG PY_EXTRA_PK0 B
94ARG MAX_JOBS=40 B
95ENV MAX_JOBS=320 B
96USER root0 B
97COPY dir:9a0662d9c5c20fa43f422b1d9f178295e34181aa791d59605dda11504da06686 in /var/lib/jenkins 6.93 KB
98COPY dir:38eb97b47ad36b235b38565e50dc70740248b6e289631881642e81d4ef15d93b in /var/lib/jenkins/pytorch 3.36 GB
99WORKDIR /var/lib/jenkins0 B
100|4 FRAME_NAME=pytorch OS_CHOICE=ubuntu24.04 PY_EXTRA_PK= PY_VER=py3.120 B
101|4 FRAME_NAME=pytorch OS_CHOICE=ubuntu24.04 PY_EXTRA_PK= PY_VER=py3.122.99 GB
102ENV ROCM_PATH=/opt/rocm0 B
103ENV MPI_HOME=/opt/ompi0 B
104ENV UCX_HOME=/opt/ucx0 B
105ENV PATH=/opt/ompi/bin:/opt/ucx/bin:/opt/cache/bin:/opt/rocm/llvm/bin:/opt/rocm/opencl/bin:/opt/rocm/hip/bin:/opt/rocm/hcc/bin:/opt/rocm/bin:/opt/conda/envs/py_3.12/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
106ENV LD_LIBRARY_PATH=/opt/ompi/lib:/opt/rocm/lib:/usr/local/lib:0 B
107ENV CI=10 B
108ENV PYTORCH_TEST_WITH_ROCM=10 B
109ENV PYTORCH_TESTING_DEVICE_ONLY_FOR=cuda0 B
110ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B
111bash2.14 MB
112ARG AITER_BRANCH=main0 B
113ARG CK_AITER_BRANCH=develop0 B
114RUN |2 AITER_BRANCH=main CK_AITER_BRANCH=develop /bin/sh4.43 GB

Command

 ARG RELEASE