genesu100/utc_ci:30962658129

Index digest

sha256:47779ac8252e4c1db7e41f1a3d7ea94342037b6f97bcdad73d148df98ff37875

OS/ARCH

linux/amd64

Compressed size

10.24 GB

Last pushed

about 1 month by genesu100

Type

Image

Manifest digest

sha256:1419301f8898197c44d9693b5fe9d4b3063ddfe66e3dc56609d5532310ef0264

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 update243.08 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

16RUN /bin/sh -c pip install32.09 MB

17ENV THE_ROCK_VERSION=7.14.00 B

18RUN /bin/sh -c python -m3.16 GB

19RUN /bin/sh -c rocm-sdk init3.16 GB

20COPY /bazelisk /usr/local/bin/bazel # buildkit5.67 MB

21ENV ROCM_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

22ENV ROCM_HOME=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

23ENV HIP_PLATFORM=amd0 B

24ENV HIP_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel0 B

25ENV HIP_CLANG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/bin0 B

26ENV HIP_INCLUDE_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B

27ENV HIP_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib0 B

28ENV HIP_DEVICE_LIB_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/llvm/amdgcn/bitcode0 B

29

ENV 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:/bin

0 B

30

ENV LD_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/lib64:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/llvm/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/host-math/lib:/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/rocm_sysdeps/lib

0 B

31

ENV 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/lib64

0 B

32ENV CPATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/include0 B

33ENV PKG_CONFIG_PATH=/opt/venv/lib/python3.12/site-packages/_rocm_sdk_devel/lib/pkgconfig0 B

34RUN /bin/sh -c git clone45.54 MB

35ARG JAX_VERSION=0.11.00 B

36ENV JAX_VERSION=0.11.00 B

37ENV JAX_WHEELHOUSE_URL=https://github.com/ROCm/rocm-jax/releases/download/rocm-jax-v0.11.0/wheelhouse_theRock7.14.zip0 B

38RUN |1 JAX_VERSION=0.11.0 /bin/sh -c285.01 MB

39ARG TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf80 B

40RUN |2 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 /bin/sh362.28 MB

41COPY /mount.gcsfuse /usr/sbin/mount.gcsfuse # buildkit18.23 MB

42ENV NVTE_ALLOW_NONDETERMINISTIC_ALGO=10 B

43ENV NVTE_USE_HIPBLASLT=10 B

44ENV GPU_MAX_HW_QUEUES=20 B

45ENV HIP_FORCE_DEV_KERNARG=10 B

46ENV HSA_FORCE_FINE_GRAIN_PCIE=10 B

47ENV NVTE_FUSED_ATTN=10 B

48ENV NCCL_DEBUG=VERSION0 B

49ENV NVTE_CK_USES_BWD_V3=10 B

50ENV NVTE_CK_USES_FWD_V3=10 B

51ENV NVTE_CK_IS_V3_ATOMIC_FP32=10 B

52ENV NVTE_CK_HOW_V3_BF16_CVT=20 B

53ENV XLA_PYTHON_CLIENT_MEM_FRACTION=.90 B

54

ENV 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=8589934592

0 B

55WORKDIR /workspace32 B

56ARG MAXTEXT_REPO=https://github.com/ROCm/maxtext.git0 B

57ARG MAXTEXT_BRANCH=release/v26.60 B

58

RUN |4 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

1.5 GB

59

RUN |4 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

253.32 MB

60ARG PRIMUS_REPO=https://github.com/AMD-AGI/Primus.git0 B

61ARG PRIMUS_BRANCH=main0 B

62

RUN |6 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

1020.08 MB

63

RUN |6 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

16.4 MB

64COPY /mount.gcsfuse /usr/sbin/mount.gcsfuse # buildkit18.23 MB

65COPY /gcsfuse /usr/bin/gcsfuse # buildkit24.84 MB

66

RUN |6 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

25.14 MB

67ARG AINIC_BUNDLE_VERSION=1.117.5-a-770 B

68

RUN |7 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

362.94 KB

69

RUN |7 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

226 B

70ARG UCX_VERSION=1.18.00 B

71

RUN |8 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

86.43 MB

72ENV UCX_INSTALL_DIR=/workspace/ucx-1.18.0/install0 B

73ARG MPI_VERSION=4.1.60 B

74

RUN |9 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

3.84 MB

75

RUN |9 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

472 B

76

RUN |9 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

32 B

77ARG GIT_COMMIT_TAG=41e867edd931e4289fb11981ea32837534b53b490 B

78ARG DOCKERFILE_PATH=jax_training_the_rock/docker/Dockerfile0 B

79

RUN |11 JAX_VERSION=0.11.0 TE_VERSION=2.15.0.dev0+rocm7.15.0a20260724.2a26bf8 MAXTEXT_REPO=https://github.com/ROCm/maxtext.git

15.21 KB

80COPY jax_training_the_rock/docker/Dockerfile /workspace/.manifest/docker-build-recipe.txt # buildkit6.34 KB

Command

 ARG RELEASE