Sign inSign up

genesu100/utc_ci:33372785601

Index digest

sha256:b99b4293319161eacb53e8fd7745621bfc9f3bfd9963c8bd1bdc877e6d219976

OS/ARCH

linux/amd64

Compressed size

10.83 GB

Last pushed

24 days by genesu100

Type

Image

Manifest digest

sha256:8e0af8f736dcc52cb1d54104d076b38403cd5a6843655d968cc348c892e6d059

Image Layers

Image Layers

1ARG RELEASE0 B
2ARG LAUNCHPAD_BUILD_ARCH0 B
3LABEL org.opencontainers.image.version=24.040 B
4ADD file ... in / 28.37 MB
5CMD ["/bin/bash"]0 B
6WORKDIR /workspace/99 B
7ENV MAX_JOBS=1280 B
8ENV PYTORCH_ROCM_ARCH=gfx942;gfx9500 B
9ENV ROCM_AMDGPU_TARGETS=gfx942,gfx9500 B
10ENV ROCPROFILER_QUEUE_INTERPOSITION=00 B
11ENV DEBUG_HIP_DYNAMIC_QUEUES=00 B
12ENV DEBIAN_FRONTEND=noninteractive0 B
13RUN /bin/sh -c apt update240.52 MB
14RUN /bin/sh -c python -m4 MB
15ENV PATH=/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
16COPY /dist/pip-*.whl /tmp/ # buildkit1.71 MB
17RUN /bin/sh -c pip install32.84 MB
18ARG ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch0 B
19ARG JAX_PJRT_PKG=jax_rocm10_pjrt0 B
20ARG JAX_PLUGIN_PKG=jax_rocm10_plugin0 B
21ARG THE_ROCK_VERSION=10.0.0rc40 B
22ENV THE_ROCK_VERSION=10.0.0rc40 B
23RUN |4 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin3.26 GB
24RUN |4 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin3.25 GB
25COPY /bazelisk /usr/local/bin/bazel # buildkit5.67 MB
26ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
27ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
28ENV HIP_PLATFORM=amd0 B
29ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B
30ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B
31ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
32ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B
33ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B
34ENV PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/bin:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin:/opt/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin0 B
35ENV LIBRARY_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib640 B
36ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B
37ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B
38ENV LD_LIBRARY_PATH=0 B
39ARG JAX_VERSION=0.11.00 B
40ENV JAX_VERSION=0.11.00 B
41ARG JAX_PJRT_VERSION=0.11.0+rocm10.0.0rc40 B
42ENV JAX_PJRT_VERSION=0.11.0+rocm10.0.0rc40 B
43ARG JAX_PLUGIN_VERSION=0.11.0+rocm10.0.0rc40 B
44ENV JAX_PLUGIN_VERSION=0.11.0+rocm10.0.0rc40 B
45RUN |7 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin285.51 MB
46RUN |7 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin2.1 KB
47ARG TE_VERSION=2.17.0+rocm10.0.0rc4.ce80fb00 B
48ARG TE_INDEX_URL=https://rocm.frameworks-prereleases.amd.com/whl-multi-arch-staging0 B
49RUN |9 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin398.86 MB
50COPY /mount.gcsfuse /usr/sbin/mount.gcsfuse # buildkit18.24 MB
51ENV NVTE_ALLOW_NONDETERMINISTIC_ALGO=10 B
52ENV NVTE_USE_HIPBLASLT=10 B
53ENV GPU_MAX_HW_QUEUES=20 B
54ENV HIP_FORCE_DEV_KERNARG=10 B
55ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B
56ENV NVTE_FUSED_ATTN=10 B
57ENV NCCL_DEBUG=VERSION0 B
58ENV NVTE_CK_USES_BWD_V3=10 B
59ENV NVTE_CK_USES_FWD_V3=10 B
60ENV NVTE_CK_IS_V3_ATOMIC_FP32=10 B
61ENV NVTE_CK_HOW_V3_BF16_CVT=20 B
62ENV XLA_PYTHON_CLIENT_MEM_FRACTION=.90 B
63ENV XLA_FLAGS=--xla_gpu_memory_limit_slop_factor=95 --xla_gpu_reduce_scatter_combine_threshold_bytes=8589934592 --xla_gpu_enable_latency_hiding_scheduler=True --xla_gpu_all_gather_combine_threshold_bytes=85899345920 B
64WORKDIR /workspace32 B
65ARG MAXTEXT_REPO=https://github.com/ROCm/maxtext.git0 B
66ARG MAXTEXT_BRANCH=release/v26.60 B
67RUN |11 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin1.5 GB
68ARG TF_REPO=https://github.com/ROCm/tensorflow-upstream.git0 B
69ARG TF_BRANCH=upstream-v2.21.00 B
70RUN |13 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin507.37 KB
71RUN |13 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin264.61 MB
72ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B
73ARG PRIMUS_BRANCH=main0 B
74RUN |15 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin1.39 GB
75RUN |15 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin20.29 MB
76COPY /mount.gcsfuse /usr/sbin/mount.gcsfuse # buildkit18.24 MB
77COPY /gcsfuse /usr/bin/gcsfuse # buildkit24.85 MB
78RUN |15 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin25.33 MB
79ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B
80RUN |16 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin363.72 KB
81RUN |16 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin227 B
82ARG UCX_VERSION=1.18.00 B
83RUN |17 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin86.43 MB
84ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B
85ARG MPI_VERSION=4.1.60 B
86RUN |18 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin3.84 MB
87RUN |18 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin535 B
88RUN |18 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin32 B
89ARG GIT_COMMIT_TAG=30ab5558b6614bb23e8ac121f021679af5facd5c0 B
90ARG DOCKERFILE_PATH=jax_training_the_rock/docker/Dockerfile0 B
91RUN |20 ROCM_INDEX_URL=https://rocm.prereleases.amd.com/whl-multi-arch JAX_PJRT_PKG=jax_rocm10_pjrt JAX_PLUGIN_PKG=jax_rocm10_plugin15.25 KB
92COPY jax_training_the_rock/docker/Dockerfile /workspace/.manifest/docker-build-recipe.txt # buildkit10.07 KB

Command

 ARG RELEASE