From: Someone Serge <redacted>
Date: Tue, 26 Dec 2023 23:34:40 +0000 (+0000)
Subject: flake.nix: rocm not yet supported on aarch64, so hide the output
X-Git-Tag: upstream/0.0.4488~2753
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=a5c088d8c698299b973d2709153e5d95295606d9;p=pkg%2Fggml%2Fsources%2Fllama.cpp

flake.nix: rocm not yet supported on aarch64, so hide the output
---

diff --git a/flake.nix b/flake.nix
index 6785b52f..920a7990 100644
--- a/flake.nix
+++ b/flake.nix
@@ -74,6 +74,7 @@
           {
             config,
             lib,
+            system,
             pkgs,
             pkgsCuda,
             pkgsRocm,
@@ -103,10 +104,12 @@
               // lib.optionalAttrs pkgs.stdenv.isLinux {
                 opencl = config.packages.default.override { useOpenCL = true; };
                 cuda = config.legacyPackages.llamaPackagesCuda.llama-cpp;
-                rocm = config.legacyPackages.llamaPackagesRocm.llama-cpp;
 
                 mpi-cpu = config.packages.default.override { useMpi = true; };
                 mpi-cuda = config.packages.default.override { useMpi = true; };
+              }
+              // lib.optionalAttrs (system == "x86_64-linux") {
+                rocm = config.legacyPackages.llamaPackagesRocm.llama-cpp;
               };
           };
       };