Sign inSign up

rocm/pytorch-xdit:v26.5

Manifest digest

sha256:b8ad9fd4b41bc116ac2aff07c1066bf369cf7fc110b1a323f6302191985a51fd

OS/ARCH

linux/amd64

Compressed size

6.03 GB

Last pushed

2 months by rocmdev

Type

Image

Manifest digest

sha256:b8ad9fd4b41bc116ac2aff07c1066bf369cf7fc110b1a323f6302191985a51fd

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.91 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/sh5.99 MB
14ARG AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle0 B
15COPY docker/rocm_runtime_artifacts.txt /tmp/ # buildkit651 B
16RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle2.3 GB
17RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle4.52 MB
18RUN |3 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle9.26 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=bc5ea32c19c604cda2f4781a97cb04d5fc4945430 B
21ARG AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle0 B
22RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle824.31 MB
23RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle3.44 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.78 KB
28COPY /ffmpeg/build/share /usr/local/share # buildkit2.98 MB
29RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle9.96 KB
30COPY docker/libkineto.conf /config/libkineto.conf # buildkit480 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 # buildkit529 B
34RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle30.36 MB
35RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle507.43 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/ # buildkit861 B
39RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle1.38 GB
40RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle10.6 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 # buildkit213.11 MB
43COPY /app /app # buildkit375.99 MB
44RUN |4 PYTHON_VERSION=default DEBIAN_FRONTEND=noninteractive AMDGPU_BUNDLE_NAME=gfx94X_gfx950_bundle14.33 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 # buildkit818 B
49COPY .ci/run.py /app/.ci/run.py # buildkit5.52 KB
50COPY benchmark_configs/xdit/ /app/.ci/benchmark_configs/ # buildkit3.53 KB
51COPY .ci/quality_check.py /app/.ci/quality_check.py # buildkit2.43 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 # buildkit58.64 KB
54ENV MIOPEN_USER_DB_PATH=/miopen_userdb0 B
55ADD references/ /app/references/ # buildkit122 B

Command

 ARG RELEASE