Sign inSign up

rocm/pytorch-xdit:v26.6

Index digest

sha256:b4b9a1176d5265cc901f1353f132a634e939f6caa91ccbc100170fac5b7882dc

OS/ARCH

linux/amd64

Compressed size

6.32 GB

Last pushed

29 days by rocmdev

Type

Image

Manifest digest

sha256:ef858f07767cd64f995380a5c339c7834cbdc5a022d913e4c2640133fa0c383b

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/sh269.81 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
14ARG AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle0 B
15COPY docker/rocm_runtime_artifacts.txt /tmp/ # buildkit713 B
16RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle2.44 GB
17RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle4.82 MB
18RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle9.3 KB
19ENV 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
20ARG AITER_COMMIT=3a5261496c591b4fd7cf280d35d20dee713098fd0 B
21ARG AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle0 B
22RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle824.3 MB
23RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle3.46 MB
24RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle7.19 MB
25COPY /ffmpeg/bin /usr/local/bin # buildkit239.69 KB
26COPY /ffmpeg/build/lib /usr/local/lib # buildkit84.71 MB
27COPY /ffmpeg/build/include /usr/local/include # buildkit289.77 KB
28COPY /ffmpeg/build/share /usr/local/share # buildkit2.98 MB
29RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle10 KB
30COPY docker/libkineto.conf /config/libkineto.conf # buildkit481 B
31ENV TRITON_HIP_LLD_PATH=/opt/rocm/lib/llvm/bin/ld.lld KINETO_CONFIG=/config/libkineto.conf0 B
32WORKDIR /app/external116 B
33COPY ./src/patches /patches # buildkit1.54 KB
34RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle30.39 MB
35RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle583.58 MB
36RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle77 B
37ARG AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle0 B
38COPY docker/rocm_devel_artifacts.txt /tmp/ # buildkit874 B
39RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle1.4 GB
40RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle10.61 KB
41ENV PATH=/opt/rocm/libexec/rocprofiler-compute:/opt/rocm/bin:/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
42COPY /venv /venv # buildkit277.34 MB
43COPY /app /app # buildkit385.08 MB
44RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle14.34 MB
45WORKDIR /app32 B
46COPY src/flux/cat.png /app/data/flux_cat.png # buildkit1 MB
47COPY src/wan_xdit/assets/i2v_input.JPG /app/data/wan_input.jpg # buildkit244.19 KB
48COPY docs/pytorch-xdit/README.md /app/README.md # buildkit816 B
49COPY .ci/run.py /app/.ci/run.py # buildkit6.33 KB
50COPY benchmark_configs/xdit/ /app/.ci/benchmark_configs/ # buildkit3.63 KB
51COPY .ci/quality_check.py /app/.ci/quality_check.py # buildkit2.44 KB
52COPY src/sweep_tooling/generate_sweep_configs.py /app/.ci/tools/generate_sweep_configs.py # buildkit1.27 KB
53COPY ./data/miopen/userdb /miopen_userdb # buildkit60.41 KB
54ENV MIOPEN_USER_DB_PATH=/miopen_userdb0 B
55ADD references/ /app/references/ # buildkit120 B

Command

 ARG RELEASE