Sign inSign up

rocm/composable_kernel:ck_pytorch

Manifest digest

sha256:ad3a4baf88e0e392c759021d4a5de6a8b011b6be532852ec63633f2c380ab038

OS/ARCH

linux/amd64

Compressed size

72.59 GB

Last pushed

3 months by illsilin

Type

Image

Manifest digest

sha256:ad3a4baf88e0e392c759021d4a5de6a8b011b6be532852ec63633f2c380ab038

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 UBUNTU_VERSION=22.040 B
8ENV DEBIAN_FRONTEND=noninteractive0 B
9ARG PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx12010 B
10ENV PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx12010 B
11COPY ./common/install_base.sh install_base.sh # buildkit1.52 KB
12RUN |2 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 /bin/sh300.05 MB
13COPY ./common/install_user.sh install_user.sh # buildkit704 B
14RUN |2 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 /bin/sh1.19 KB
15ARG KATEX=0 B
16COPY ./common/install_docs_reqs.sh install_docs_reqs.sh # buildkit483 B
17RUN |3 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=108 B
18ARG ANACONDA_PYTHON_VERSION=3.100 B
19ARG BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm7.2-py3.100 B
20ENV ANACONDA_PYTHON_VERSION=3.100 B
21ENV PATH=/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
22COPY requirements-ci.txt /opt/conda/requirements-ci.txt # buildkit4.33 KB
23COPY ./common/install_conda.sh install_conda.sh # buildkit1.69 KB
24COPY ./common/common_utils.sh common_utils.sh # buildkit723 B
25RUN |5 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=3.09 GB
26ARG GCC_VERSION=0 B
27COPY ./common/install_gcc.sh install_gcc.sh # buildkit396 B
28RUN |6 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=104 B
29ARG CLANG_VERSION=0 B
30COPY ./common/install_clang.sh install_clang.sh # buildkit788 B
31RUN |7 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=106 B
32ARG VISION=yes0 B
33COPY ./common/install_vision.sh ./common/cache_vision_models.sh ./common/common_utils.sh ./1.46 KB
34RUN |8 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=442.03 MB
35RUN |8 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=163 B
36ENV INSTALLED_VISION=yes0 B
37ARG ROCM_VERSION=7.20 B
38ENV ROCM_VERSION=7.20 B
39RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=104 B
40COPY ./common/common_utils.sh common_utils.sh # buildkit723 B
41COPY ./ci_commit_pins/rocm-composable-kernel.txt ci_commit_pins/rocm-composable-kernel.txt # buildkit196 B
42COPY ./common/install_rocm.sh install_rocm.sh # buildkit3.98 KB
43RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=8.99 GB
44RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=135 B
45RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=104 B
46COPY ./common/install_rocm_magma.sh install_rocm_magma.sh # buildkit609 B
47RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=830.67 MB
48RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=111 B
49ADD ./common/install_miopen.sh install_miopen.sh # buildkit1.52 KB
50RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=107 B
51ADD ./common/install_rocm_drm.sh install_rocm_drm.sh # buildkit1.7 KB
52RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=66.32 MB
53ENV LANG=C.UTF-80 B
54ENV LC_ALL=C.UTF-80 B
55COPY ./common/install_amdsmi.sh install_amdsmi.sh # buildkit332 B
56RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=3.23 MB
57RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=107 B
58COPY ./common/install_openssl.sh install_openssl.sh # buildkit424 B
59ENV OPENSSL_ROOT_DIR=/opt/openssl0 B
60RUN |9 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=18.41 MB
61ENV OPENSSL_DIR=/opt/openssl0 B
62ARG INDUCTOR_BENCHMARKS=0 B
63ARG ANACONDA_PYTHON_VERSION=3.100 B
64ENV ANACONDA_PYTHON_VERSION=3.100 B
65COPY ./common/install_inductor_benchmark_deps.sh install_inductor_benchmark_deps.sh # buildkit1.04 KB
66COPY ./common/common_utils.sh common_utils.sh # buildkit723 B
67COPY ci_commit_pins/huggingface-requirements.txt huggingface-requirements.txt # buildkit149 B
68COPY ci_commit_pins/timm.txt timm.txt # buildkit137 B
69COPY ci_commit_pins/torchbench.txt torchbench.txt # buildkit140 B
70RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=32 B
71RUN |11 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=222 B
72ARG NINJA_VERSION=1.9.00 B
73COPY ./common/install_ninja.sh install_ninja.sh # buildkit346 B
74RUN |12 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=86.24 KB
75RUN |12 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=106 B
76ARG TRITON=yes0 B
77COPY ./common/install_triton.sh install_triton.sh # buildkit1.63 KB
78COPY ./common/common_utils.sh common_utils.sh # buildkit723 B
79COPY ci_commit_pins/triton.txt triton.txt # buildkit138 B
80COPY triton_version.txt triton_version.txt # buildkit118 B
81RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=7.95 GB
82RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=174 B
83COPY ./common/install_cache.sh install_cache.sh # buildkit2.1 KB
84ENV PATH=/opt/cache/bin:/opt/conda/envs/py_3.10/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
85RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=359.7 MB
86COPY ./common/install_openmpi.sh install_openmpi.sh # buildkit303 B
87RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=32 B
88RUN |13 UBUNTU_VERSION=22.04 PYTORCH_ROCM_ARCH=gfx906;gfx908;gfx90a;gfx942;gfx950;gfx1030;gfx1100;gfx1101;gfx1102;gfx1150;gfx1151;gfx1200;gfx1201 KATEX=108 B
89ARG BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm7.2-py3.100 B
90ENV BUILD_ENVIRONMENT=pytorch-linux-jammy-rocm7.2-py3.100 B
91COPY /opt/llvm /opt/llvm # buildkit51.64 MB
92USER jenkins0 B
93CMD ["bash"]0 B
94WORKDIR /tmp32 B
95USER root0 B
96ENV CI=10 B
97ENV PYTORCH_TEST_WITH_ROCM=10 B
98ENV PYTORCH_TESTING_DEVICE_ONLY_FOR=cuda0 B
99RUN /bin/sh -c git clone13.57 GB
100RUN /bin/sh -c git clone1.26 GB
101RUN /bin/sh -c git clone775.01 MB
102ARG CK_PYTORCH_BRANCH=develop0 B
103RUN |1 CK_PYTORCH_BRANCH=develop /bin/sh -c34.91 GB

Command

 ARG RELEASE