ROCm 6 (#96)

---------

Co-authored-by: Felix Thaler <[email protected]>

.github/workflows/build.yml

@@ -142,20 +142,26 @@ jobs:
 
   hip:
     runs-on: ubuntu-latest
-    needs: [base, parmetis]
+    needs: [base]
     strategy:
       matrix:
-        base: [base, gcc-9-ucx-mpi-atlas-parmetis]
+        base: [base]
     steps:
     - uses: actions/checkout@v2
+    - name: Free Disk Space (Ubuntu)
+      uses: jlumbroso/free-disk-space@main
+      with:
+        # this might remove tools that are actually needed,
+        # if set to "true" but frees about 6 GB
+        tool-cache: false
     - name: Build
       run: >
         docker build
         --progress=plain
         --cache-from $CR_REPOSITORY:${{ matrix.base }}
-        --cache-from $CR_REPOSITORY:${{ matrix.base }}-hip
+        --cache-from $CR_REPOSITORY:${{ matrix.base }}-rocm-6.2
         --build-arg BUILDKIT_INLINE_CACHE=1
-        --tag $CR_REPOSITORY:${{ matrix.base }}-hip
+        --tag $CR_REPOSITORY:${{ matrix.base }}-rocm-6.2
         --build-arg REPOSITORY=$CR_REPOSITORY
         --build-arg BASE=${{ matrix.base }}
         hip

hip/Dockerfile

@@ -3,14 +3,11 @@ ARG BASE
 FROM ${REPOSITORY}:${BASE}
 LABEL maintainer="Felix Thaler <[email protected]>"
 
-# from https://docs.amd.com/bundle/ROCm-Installation-Guide-v5.4/page/How_to_Install_ROCm.html
-# TODO(havogt): remove jammy packages once official support for 24.04 is there,
-# workaround taken from https://github.com/nktice/AMD-AI/blob/99b51b7ce08f339e81ccede05d65a21d10c6a27b/README.md
-RUN add-apt-repository -y -s deb http://security.ubuntu.com/ubuntu jammy main universe && \
-    add-apt-repository ppa:deadsnakes/ppa -y && \
-    wget https://repo.radeon.com/amdgpu-install/22.40/ubuntu/jammy/amdgpu-install_5.4.50401-1_all.deb && \
+# NOTE: If you change the ROCm version here, also change it in the build.yml to avoid overwriting the old image,
+# see https://github.com/GridTools/gridtools-docker/issues/97.
+RUN wget https://repo.radeon.com/amdgpu-install/6.2.1/ubuntu/noble/amdgpu-install_6.2.60201-1_all.deb && \
     apt-get update -qq && \
-    DEBIAN_FRONTEND=noninteractive apt-get install -qq -y ./amdgpu-install_5.4.50401-1_all.deb && \
+    DEBIAN_FRONTEND=noninteractive apt-get install -qq -y ./amdgpu-install_6.2.60201-1_all.deb && \
     apt-get update -qq && \
     amdgpu-install -y --usecase=rocm,hip --no-dkms && \
     rm -rf /var/lib/apt/lists/*