Sign inSign up

vllm/vllm-omni-rocm:v0.20.0

Manifest digest

sha256:42292286e8064bdb940c129f02246aca129422ade143c9d254a180d6982c46f5

OS/ARCH

linux/amd64

Compressed size

10.5 GB

Last pushed

5 months by vllmbot

Type

Image

Manifest digest

sha256:42292286e8064bdb940c129f02246aca129422ade143c9d254a180d6982c46f5

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 / 29.04 MB
6CMD ["/bin/bash"]0 B
7LABEL [email protected]0 B
8ARG ROCM_VERSION=5.30 B
9ARG AMDGPU_VERSION=5.30 B
10ARG APT_PREF0 B
11RUN |3 ROCM_VERSION=7.2.1 AMDGPU_VERSION=30.30.1 APT_PREF=Package:239 B
12RUN |3 ROCM_VERSION=7.2.1 AMDGPU_VERSION=30.30.1 APT_PREF=Package:6.76 GB
13RUN |3 ROCM_VERSION=7.2.1 AMDGPU_VERSION=30.30.1 APT_PREF=Package:654 B
14ENV PATH=/opt/rocm/llvm/bin:/opt/rocm/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
15ENV ROCM_PATH=/opt/rocm0 B
16ENV LD_LIBRARY_PATH=/opt/rocm/lib:/usr/local/lib:0 B
17ARG PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
18ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
19ENV AITER_ROCM_ARCH=gfx942;gfx9500 B
20ENV MORI_GPU_ARCHS=gfx942;gfx9500 B
21ENV HSA_NO_SCRATCH_RECLAIM=10 B
22ARG PYTHON_VERSION=3.120 B
23ENV PYTHON_VERSION=3.120 B
24RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh93 B
25WORKDIR /app32 B
26ENV DEBIAN_FRONTEND=noninteractive0 B
27RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh147.27 MB
28RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh65.91 MB
29RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh4.06 MB
30ARG USE_SCCACHE0 B
31ARG SCCACHE_DOWNLOAD_URL0 B
32ARG SCCACHE_ENDPOINT0 B
33ARG SCCACHE_BUCKET_NAME0 B
34ARG SCCACHE_REGION_NAME0 B
35ARG SCCACHE_S3_NO_CREDENTIALS0 B
36RUN |8 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 USE_SCCACHE=19.11 MB
37RUN |8 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 USE_SCCACHE=1227 B
38ARG USE_SCCACHE0 B
39ENV SCCACHE_BUCKET=vllm-build-sccache0 B
40ENV SCCACHE_REGION=us-west-20 B
41ENV SCCACHE_S3_NO_CREDENTIALS=00 B
42ENV SCCACHE_IDLE_TIMEOUT=00 B
43RUN /bin/sh -c pip install2.03 GB
44ARG BASE_IMAGE0 B
45ARG TRITON_BRANCH0 B
46ARG TRITON_REPO0 B
47ARG PYTORCH_BRANCH0 B
48ARG PYTORCH_VISION_BRANCH0 B
49ARG PYTORCH_REPO0 B
50ARG PYTORCH_VISION_REPO0 B
51ARG PYTORCH_AUDIO_BRANCH0 B
52ARG PYTORCH_AUDIO_REPO0 B
53ARG FA_BRANCH0 B
54ARG FA_REPO0 B
55ARG AITER_BRANCH0 B
56ARG AITER_REPO0 B
57ARG MORI_BRANCH0 B
58ARG MORI_REPO0 B
59RUN |15 BASE_IMAGE=rocm/dev-ubuntu-22.04:7.2.1-complete TRITON_BRANCH=ba5c1517 TRITON_REPO=https://github.com/ROCm/triton.git396 B
60ARG ARG_PYTORCH_ROCM_ARCH0 B
61ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
62RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c137.45 MB
63RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c5.27 MB
64ARG USE_SCCACHE0 B
65RUN |2 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 /bin/sh32 B
66RUN |2 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 /bin/sh23.66 MB
67ENV UV_HTTP_TIMEOUT=5000 B
68ENV UV_INDEX_STRATEGY=unsafe-best-match0 B
69ENV UV_LINK_MODE=copy0 B
70ARG USE_SCCACHE0 B
71ARG SCCACHE_DOWNLOAD_URL0 B
72ARG SCCACHE_ENDPOINT0 B
73ARG SCCACHE_BUCKET_NAME0 B
74ARG SCCACHE_REGION_NAME0 B
75ARG SCCACHE_S3_NO_CREDENTIALS0 B
76RUN |8 ARG_PYTORCH_ROCM_ARCH= USE_SCCACHE=1 USE_SCCACHE=132 B
77ARG USE_SCCACHE0 B
78ENV SCCACHE_BUCKET=vllm-build-sccache0 B
79ENV SCCACHE_REGION=us-west-20 B
80ENV SCCACHE_S3_NO_CREDENTIALS=00 B
81ENV SCCACHE_IDLE_TIMEOUT=00 B
82ARG COMMON_WORKDIR0 B
83WORKDIR /app32 B
84ARG NIC_BACKEND0 B
85ARG AINIC_VERSION0 B
86ARG UBUNTU_CODENAME0 B
87RUN |3 NIC_BACKEND=none AINIC_VERSION=1.117.5 UBUNTU_CODENAME=jammy14.19 MB
88RUN /bin/sh -c python3 -m2 KB
89RUN /bin/sh -c rm -f197 B
90ENV SCCACHE_BUCKET=0 B
91ENV SCCACHE_REGION=0 B
92ENV SCCACHE_S3_NO_CREDENTIALS=0 B
93ENV SCCACHE_IDLE_TIMEOUT=0 B
94RUN /bin/sh -c case "$(which32 B
95RUN /bin/sh -c uv pip6.25 MB
96RUN /bin/sh -c cd /install692.41 MB
97ARG COMMON_WORKDIR0 B
98ARG BASE_IMAGE0 B
99ARG NIC_BACKEND0 B
100ARG AINIC_VERSION0 B
101COPY /benchmarks /app/vllm/benchmarks # buildkit223.63 KB
102COPY /examples /app/vllm/examples # buildkit241.1 KB
103COPY /docker /app/vllm/docker # buildkit6.69 KB
104ENV TOKENIZERS_PARALLELISM=false0 B
105ENV SAFETENSORS_FAST_GPU=10 B
106ENV HIP_FORCE_DEV_KERNARG=10 B
107RUN |4 COMMON_WORKDIR=/app BASE_IMAGE=public.ecr.aws/q9t5s3a7/vllm-release-repo:11c6f3c16ed78ff5-rocm-base NIC_BACKEND=none168 B
108ENV KINETO_CONFIG=/app/libkineto.conf0 B
109RUN |4 COMMON_WORKDIR=/app BASE_IMAGE=public.ecr.aws/q9t5s3a7/vllm-release-repo:11c6f3c16ed78ff5-rocm-base NIC_BACKEND=none487 B
110CMD ["/bin/bash"]0 B
111ENTRYPOINT ["vllm" "serve"]0 B
112ARG USE_NIGHTLY_BUILD=00 B
113ARG VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab0 B
114ARG ARG_PYTORCH_ROCM_ARCH0 B
115ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
116ARG COMMON_WORKDIR=/app0 B
117WORKDIR /app32 B
118RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=112.67 MB
119RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=32 B
120RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=118 B
121COPY . /app/vllm-omni # buildkit42.68 MB
122RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=289 B
123RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=464.12 MB
124RUN |4 USE_NIGHTLY_BUILD=0 VLLM_VERSION_OR_COMMIT_HASH=89138b21cc246ae944c741d5c399c148e2b770ab ARG_PYTORCH_ROCM_ARCH=149 B
125ENTRYPOINT ["vllm" "serve" "--omni"]0 B
126ENV MIOPEN_DEBUG_CONV_DIRECT=00 B
127ENV MIOPEN_DEBUG_CONV_GEMM=00 B
128COPY ./vllmomni0200/vllm_omni/diffusion/layers/rope.py /usr/local/lib/python3.12/dist-packages/vllm_omni/diffusion/layers/rope.py # buildkit1.88 KB

Command

 ARG RELEASE