Sign inSign up

amdenterpriseai/aim-base:0.9

Manifest digest

sha256:4d1908a0f3918c0fd1297101956bc17daaa41af5445f84c19b700e38577e872c

OS/ARCH

linux/amd64

Compressed size

8.86 GB

Last pushed

10 months by andyallred

Type

Image

Manifest digest

sha256:4d1908a0f3918c0fd1297101956bc17daaa41af5445f84c19b700e38577e872c

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
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.0 AMDGPU_VERSION=7.0 APT_PREF=Package:237 B
12RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:4.99 GB
13RUN |3 ROCM_VERSION=7.0 AMDGPU_VERSION=7.0 APT_PREF=Package:656 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
20ARG PYTHON_VERSION=3.120 B
21RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh93 B
22WORKDIR /app32 B
23ENV DEBIAN_FRONTEND=noninteractive0 B
24RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh112.15 MB
25RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh65.7 MB
26RUN |2 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 /bin/sh2.1 GB
27ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete0 B
28ARG TRITON_BRANCH=57c693b60 B
29ARG TRITON_REPO=https://github.com/ROCm/triton.git0 B
30ARG PYTORCH_BRANCH=1c57644d0 B
31ARG PYTORCH_VISION_BRANCH=v0.23.00 B
32ARG PYTORCH_REPO=https://github.com/ROCm/pytorch.git0 B
33ARG PYTORCH_VISION_REPO=https://github.com/pytorch/vision.git0 B
34ARG FA_BRANCH=0e60e3940 B
35ARG FA_REPO=https://github.com/Dao-AILab/flash-attention.git0 B
36ARG AITER_BRANCH=9716b1b80 B
37ARG AITER_REPO=https://github.com/ROCm/aiter.git0 B
38RUN |13 PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx1151 PYTHON_VERSION=3.12 BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete350 B
39COPY pytorch_afo_testkit/afo/gemm/benchmarks/afo_tune_device_*_full.csv /app # buildkit61.21 MB
40ENV PYTORCH_TUNABLEOP_ENABLED=10 B
41ENV PYTORCH_TUNABLEOP_TUNING=00 B
42ENV PYTORCH_TUNABLEOP_FILENAME=/app/afo_tune_device_%d_full.csv0 B
43ARG ARG_PYTORCH_ROCM_ARCH0 B
44ENV PYTORCH_ROCM_ARCH=gfx90a;gfx942;gfx950;gfx1100;gfx1101;gfx1200;gfx1201;gfx1150;gfx11510 B
45RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c94.55 MB
46RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c1.35 KB
47RUN |1 ARG_PYTORCH_ROCM_ARCH= /bin/sh -c217 B
48ARG COMMON_WORKDIR=/app0 B
49WORKDIR /app32 B
50RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh1.97 KB
51RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh32 B
52RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh13.37 MB
53RUN |2 ARG_PYTORCH_ROCM_ARCH= COMMON_WORKDIR=/app /bin/sh1.26 GB
54ARG COMMON_WORKDIR=/app0 B
55COPY /benchmarks /app/vllm/benchmarks # buildkit149.94 KB
56COPY /examples /app/vllm/examples # buildkit177.52 KB
57COPY /docker /app/vllm/docker # buildkit1.63 KB
58ENV RAY_EXPERIMENTAL_NOSET_ROCR_VISIBLE_DEVICES=10 B
59ENV RAY_EXPERIMENTAL_NOSET_HIP_VISIBLE_DEVICES=10 B
60ENV TOKENIZERS_PARALLELISM=false0 B
61ENV SAFETENSORS_FAST_GPU=10 B
62ENV HIP_FORCE_DEV_KERNARG=10 B
63CMD ["/bin/bash"]0 B
64ARG BASE_REGISTRY_NAMESPACE=rocm0 B
65ARG BASE_REPOSITORY=vllm0 B
66ARG BASE_TAG=rocm7.0.0_vllm_0.11.1_202511030 B
67ARG INSTALL_LMCACHE=true0 B
68ARG INSTALL_LMCACHE_COMMIT=d5be14f0 B
69ENV PYTHONUNBUFFERED=10 B
70ENV DEBIAN_FRONTEND=noninteractive0 B
71ENV PYTHONPATH=/workspace/aim-runtime/src0 B
72ENV HF_HOME=/workspace/model-cache0 B
73ENV HF_DATASETS_CACHE=/workspace/model-cache0 B
74ENV HF_HUB_DISABLE_TELEMETRY=10 B
75ENV AIM_BASE_IMAGE_REF=rocm/vllm:rocm7.0.0_vllm_0.11.1_202511030 B
76ENV AIM_ALLOW_GENERAL_PROFILE_FALLBACK=true0 B
77RUN |5 BASE_REGISTRY_NAMESPACE=rocm BASE_REPOSITORY=vllm BASE_TAG=rocm7.0.0_vllm_0.11.1_2025110332 B
78WORKDIR /workspace99 B
79COPY src/ /workspace/aim-runtime/src/ # buildkit104.49 KB
80COPY pyproject.toml /workspace/aim-runtime/ # buildkit1.48 KB
81COPY requirements/requirements.txt /workspace/aim-runtime/ # buildkit578 B
82COPY src/entrypoint.py /workspace/entrypoint.py # buildkit2.52 KB
83RUN |5 BASE_REGISTRY_NAMESPACE=rocm BASE_REPOSITORY=vllm BASE_TAG=rocm7.0.0_vllm_0.11.1_202511032.52 KB
84COPY profiles/general /workspace/aim-runtime/profiles/general # buildkit875 B
85COPY schemas /workspace/aim-runtime/schemas # buildkit4.36 KB
86RUN |5 BASE_REGISTRY_NAMESPACE=rocm BASE_REPOSITORY=vllm BASE_TAG=rocm7.0.0_vllm_0.11.1_2025110310.94 MB
87RUN |5 BASE_REGISTRY_NAMESPACE=rocm BASE_REPOSITORY=vllm BASE_TAG=rocm7.0.0_vllm_0.11.1_20251103125.79 MB
88RUN |5 BASE_REGISTRY_NAMESPACE=rocm BASE_REPOSITORY=vllm BASE_TAG=rocm7.0.0_vllm_0.11.1_20251103126 B
89ENTRYPOINT ["./entrypoint.py"]0 B

Command

 ARG RELEASE