fix cmake errors on fedora

Signed-off-by: Adam Miller <[email protected]>
containers · Jan 28, 2025 · f90a99c · f90a99c
1 parent a4ac319
commit f90a99c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/container-images/scripts/build_llama_and_whisper.sh b/container-images/scripts/build_llama_and_whisper.sh
@@ -37,7 +37,7 @@ dnf_install() {
     dnf install -y mesa-vulkan-drivers "${vulkan_rpms[@]}" "${rpm_list[@]}"
   elif [ "$containerfile" = "rocm" ]; then
     if [ "${ID}" = "fedora" ]; then
-      dnf install -y rocm-core-devel hipblas-devel rocblas-devel
+      dnf install -y rocm-core-devel hipblas-devel rocblas-devel rocm-hip-devel
     else
       dnf install -y rocm-dev hipblas-devel rocblas-devel
     fi
@@ -72,6 +72,9 @@ configure_common_flags() {
   case "$containerfile" in
     rocm)
       common_flags+=("-DGGML_HIP=ON" "-DAMDGPU_TARGETS=${AMDGPU_TARGETS:-gfx1010,gfx1030,gfx1032,gfx1100,gfx1101,gfx1102}")
+      if [ "${ID}" = "fedora" ]; then
+        common_flags+=("-DCMAKE_HIP_COMPILER_ROCM_ROOT=/usr")
+      fi
       ;;
     cuda)
       common_flags+=("-DGGML_CUDA=ON" "-DCMAKE_EXE_LINKER_FLAGS=-Wl,--allow-shlib-undefined")