llama-cpp: allow to override ROCM GPU targets (#435263)

This commit is contained in:
Philip Taron
2025-08-23 09:28:44 -07:00
committed by GitHub
+2 -2
View File
@@ -12,6 +12,7 @@
rocmSupport ? config.rocmSupport,
rocmPackages ? { },
rocmGpuTargets ? builtins.concatStringsSep ";" rocmPackages.clr.gpuTargets,
openclSupport ? false,
clblast,
@@ -146,8 +147,7 @@ effectiveStdenv.mkDerivation (finalAttrs: {
]
++ optionals rocmSupport [
(cmakeFeature "CMAKE_HIP_COMPILER" "${rocmPackages.clr.hipClangPath}/clang++")
# TODO: this should become `clr.gpuTargets` in the future.
(cmakeFeature "CMAKE_HIP_ARCHITECTURES" rocmPackages.rocblas.amdgpu_targets)
(cmakeFeature "CMAKE_HIP_ARCHITECTURES" rocmGpuTargets)
]
++ optionals metalSupport [
(cmakeFeature "CMAKE_C_FLAGS" "-D__ARM_FEATURE_DOTPROD=1")