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

1 files changed, 4 insertions, 1 deletions

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;
               };
           };
       };