Sign inSign up

rocm/pytorch-xdit:v26.7

Index digest

sha256:b52c9189f4661958ca24df5266c931a2f243c923d1bd835978b174ab88f304d1

OS/ARCH

linux/amd64

Compressed size

6.47 GB

Last pushed

7 days by rocm

Type

Image

Manifest digest

sha256:dd02f8d33a2b05c16ddb815425ec634d911753a3d8a95b08dd46ff259dbad7b9

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
6ARG PYTHON_VERSION=default0 B
7ARG DEBIAN_FRONTEND=noninteractive0 B
8RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh272.07 MB
9COPY /opt/python/usr /usr # buildkit32 B
10RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh19.73 MB
11RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh4 MB
12ENV PATH=/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
13RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh6.08 MB
14COPY docker/rocm_runtime_artifacts.txt /tmp/ # buildkit737 B
15RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh2.54 GB
16RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh5.03 MB
17RUN |2 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive /bin/sh9.22 KB
18ENV PATH=/opt/rocm/bin:/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin ROCM_HOME=/opt/rocm ROCM_PATH=/opt/rocm HIP_DEVICE_LIB_PATH=/opt/rocm/lib/llvm/amdgcn/bitcode0 B
19ARG AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd0 B
20RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd823.95 MB
21RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd3.56 MB
22RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd8.14 MB
23COPY /ffmpeg/bin /usr/local/bin # buildkit239.68 KB
24COPY /ffmpeg/build/lib /usr/local/lib # buildkit84.71 MB
25COPY /ffmpeg/build/include /usr/local/include # buildkit289.77 KB
26COPY /ffmpeg/build/share /usr/local/share # buildkit2.98 MB
27RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd9.98 KB
28COPY docker/libkineto.conf /config/libkineto.conf # buildkit481 B
29ENV TRITON_HIP_LLD_PATH=/opt/rocm/lib/llvm/bin/ld.lld KINETO_CONFIG=/config/libkineto.conf0 B
30WORKDIR /app/external118 B
31COPY ./src/code_objects /code_objects # buildkit8.02 KB
32ENV AITER_USE_SYSTEM_TRITON=10 B
33RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd666.72 MB
34COPY docker/rocm_devel_artifacts.txt /tmp/ # buildkit882 B
35RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd1.39 GB
36RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd10.43 KB
37ENV PATH=/opt/rocm/libexec/rocprofiler-compute:/opt/rocm/bin:/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
38COPY /venv /venv # buildkit274.9 MB
39COPY /app /app # buildkit397.13 MB
40RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd14.37 MB
41WORKDIR /app32 B
42COPY src/flux/cat.png /app/data/flux_cat.png # buildkit1 MB
43COPY src/wan_xdit/assets/i2v_input.JPG /app/data/wan_input.jpg # buildkit244.19 KB
44COPY docs/pytorch-xdit/README.md /app/README.md # buildkit817 B
45COPY .ci/run.py /app/.ci/run.py # buildkit6.65 KB
46COPY benchmark_configs/xdit/ /app/.ci/benchmark_configs/ # buildkit5.58 KB
47COPY .ci/quality_check.py /app/.ci/quality_check.py # buildkit2.43 KB
48COPY src/sweep_tooling/generate_sweep_configs.py /app/.ci/tools/generate_sweep_configs.py # buildkit1.27 KB
49COPY ./data/miopen/userdb /miopen_userdb # buildkit71.87 KB
50ENV MIOPEN_USER_DB_PATH=/miopen_userdb0 B
51ADD references/ /app/references/ # buildkit121 B

Command

 ARG RELEASE