Skip to content

Commit 3571565

Browse files
authored
Update ROCm docker container to 7.2 release (ggml-org#19418)
Also update architectures
1 parent f75c4e8 commit 3571565

1 file changed

Lines changed: 6 additions & 7 deletions

File tree

.devops/rocm.Dockerfile

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
ARG UBUNTU_VERSION=24.04
22

33
# This needs to generally match the container host's environment.
4-
ARG ROCM_VERSION=7.0
5-
ARG AMDGPU_VERSION=7.0
4+
ARG ROCM_VERSION=7.2
5+
ARG AMDGPU_VERSION=7.2
66

77
# Target the ROCm build image
88
ARG BASE_ROCM_DEV_CONTAINER=rocm/dev-ubuntu-${UBUNTU_VERSION}:${ROCM_VERSION}-complete
@@ -11,13 +11,12 @@ ARG BASE_ROCM_DEV_CONTAINER=rocm/dev-ubuntu-${UBUNTU_VERSION}:${ROCM_VERSION}-co
1111
FROM ${BASE_ROCM_DEV_CONTAINER} AS build
1212

1313
# Unless otherwise specified, we make a fat build.
14-
# List from https://github.com/ggml-org/llama.cpp/pull/1087#issuecomment-1682807878
1514
# This is mostly tied to rocBLAS supported archs.
16-
# gfx803, gfx900, gfx906, gfx1032, gfx1101, gfx1102,not officialy supported
17-
# check https://rocm.docs.amd.com/projects/install-on-linux/en/docs-6.4.1/reference/system-requirements.html
15+
# check https://rocm.docs.amd.com/projects/install-on-linux/en/docs-7.2.0/reference/system-requirements.html
16+
# check https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/compatibility/compatibilityrad/native_linux/native_linux_compatibility.html
17+
# check https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/compatibility/compatibilityryz/native_linux/native_linux_compatibility.html
1818

19-
ARG ROCM_DOCKER_ARCH='gfx803;gfx900;gfx906;gfx908;gfx90a;gfx942;gfx1010;gfx1030;gfx1032;gfx1100;gfx1101;gfx1102;gfx1200;gfx1201;gfx1151'
20-
#ARG ROCM_DOCKER_ARCH='gfx1151'
19+
ARG ROCM_DOCKER_ARCH='gfx908;gfx90a;gfx942;gfx1030;gfx1100;gfx1101;gfx1151;gfx1150;gfx1200;gfx1201'
2120

2221
# Set ROCm architectures
2322
ENV AMDGPU_TARGETS=${ROCM_DOCKER_ARCH}

0 commit comments

Comments
 (0)