From: Adam Date: Sun, 14 Sep 2025 18:43:54 +0000 (+1000) Subject: rocm.Dockerfile: added gfx1200,gfx1201 architectures to support AMD Radeon RX 9000... X-Git-Tag: upstream/0.0.6527~55 X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=0fa154e3502e940df914f03b41475a2b80b985b0;p=pkg%2Fggml%2Fsources%2Fllama.cpp rocm.Dockerfile: added gfx1200,gfx1201 architectures to support AMD Radeon RX 9000 series (#15994) * rocm.Dockerfile: added gfx1200,gfx1201 architectures to support AMD Radeon RX 9000 series https://rocm.docs.amd.com/projects/install-on-linux/en/docs-6.4.1/reference/system-requirements.html#rdna-os states the Radeon RX 9000 series is supported support from Ubuntu 24.04.2, and the dockerfile is using 24.04 which is ROCm 6.4. This fixed the `ROCm error: invalid device function` I was getting when trying to use the rocm container. --- diff --git a/.devops/rocm.Dockerfile b/.devops/rocm.Dockerfile index cf19e6e0..22107719 100644 --- a/.devops/rocm.Dockerfile +++ b/.devops/rocm.Dockerfile @@ -4,7 +4,7 @@ ARG UBUNTU_VERSION=24.04 ARG ROCM_VERSION=6.4 ARG AMDGPU_VERSION=6.4 -# Target the CUDA build image +# Target the ROCm build image ARG BASE_ROCM_DEV_CONTAINER=rocm/dev-ubuntu-${UBUNTU_VERSION}:${ROCM_VERSION}-complete ### Build image @@ -15,12 +15,12 @@ FROM ${BASE_ROCM_DEV_CONTAINER} AS build # This is mostly tied to rocBLAS supported archs. # gfx803, gfx900, gfx1032, gfx1101, gfx1102,not officialy supported # gfx906 is deprecated -#check https://rocm.docs.amd.com/projects/install-on-linux/en/docs-6.2.4/reference/system-requirements.html +#check https://rocm.docs.amd.com/projects/install-on-linux/en/docs-6.4.1/reference/system-requirements.html -ARG ROCM_DOCKER_ARCH='gfx803,gfx900,gfx906,gfx908,gfx90a,gfx942,gfx1010,gfx1030,gfx1032,gfx1100,gfx1101,gfx1102' +ARG ROCM_DOCKER_ARCH='gfx803,gfx900,gfx906,gfx908,gfx90a,gfx942,gfx1010,gfx1030,gfx1032,gfx1100,gfx1101,gfx1102,gfx1200,gfx1201' #ARG ROCM_DOCKER_ARCH=gfx1100 -# Set nvcc architectured +# Set ROCm architectured ENV AMDGPU_TARGETS=${ROCM_DOCKER_ARCH} # Enable ROCm # ENV CC=/opt/rocm/llvm/bin/clang