From: Evgeny Kurnevsky <redacted>
Date: Sat, 14 Dec 2024 18:17:36 +0000 (+0000)
Subject: nix: allow to override rocm gpu targets (#10794)
X-Git-Tag: upstream/0.0.4488~160
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=e52aba537a34d51a65cddec6bc6dafc9031edc63;p=pkg%2Fggml%2Fsources%2Fllama.cpp

nix: allow to override rocm gpu targets (#10794)

This allows to reduce compile time when you are building for a single GPU.
---

diff --git a/.devops/nix/package.nix b/.devops/nix/package.nix
index b88e6ca8..043c4364 100644
--- a/.devops/nix/package.nix
+++ b/.devops/nix/package.nix
@@ -31,6 +31,7 @@
   # Increases the runtime closure size by ~700M
   useMpi ? false,
   useRocm ? config.rocmSupport,
+  rocmGpuTargets ? builtins.concatStringsSep ";" rocmPackages.clr.gpuTargets,
   enableCurl ? true,
   useVulkan ? false,
   llamaVersion ? "0.0.0", # Arbitrary version, substituted by the flake
@@ -188,7 +189,7 @@ effectiveStdenv.mkDerivation (finalAttrs: {
     ]
     ++ optionals useRocm [
       (cmakeFeature "CMAKE_HIP_COMPILER" "${rocmPackages.llvm.clang}/bin/clang")
-      (cmakeFeature "CMAKE_HIP_ARCHITECTURES" (builtins.concatStringsSep ";" rocmPackages.clr.gpuTargets))
+      (cmakeFeature "CMAKE_HIP_ARCHITECTURES" rocmGpuTargets)
     ]
     ++ optionals useMetalKit [
       (lib.cmakeFeature "CMAKE_C_FLAGS" "-D__ARM_FEATURE_DOTPROD=1")