Skip to content

Commit

Permalink
ROCm 6 (#96)
Browse files Browse the repository at this point in the history
---------

Co-authored-by: Felix Thaler <felix.thaler@nummi.ch>
  • Loading branch information
havogt and fthaler authored Sep 25, 2024
1 parent 400fc66 commit f1f1df3
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 11 deletions.
14 changes: 10 additions & 4 deletions .github/workflows/build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
11 changes: 4 additions & 7 deletions hip/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,11 @@ ARG BASE
FROM ${REPOSITORY}:${BASE}
LABEL maintainer="Felix Thaler <thaler@cscs.ch>"

# 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/*
Expand Down

0 comments on commit f1f1df3

Please sign in to comment.