From: Someone Serge 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; }; }; };