create single function to optimize for system
This commit is contained in:
parent
b764d2de45
commit
3653e06c7d
10
flake.nix
10
flake.nix
@ -135,7 +135,7 @@
|
||||
{
|
||||
formatter.x86_64-linux = nixpkgs.legacyPackages.x86_64-linux.nixfmt-rfc-style;
|
||||
nixosConfigurations.${hostname} = nixpkgs.lib.nixosSystem {
|
||||
specialArgs = {
|
||||
specialArgs = rec {
|
||||
inherit
|
||||
username
|
||||
hostname
|
||||
@ -160,6 +160,14 @@
|
||||
# stdenv = pkgs.clang19Stdenv;
|
||||
}
|
||||
);
|
||||
|
||||
optimizePackage =
|
||||
pkg:
|
||||
optimizeWithFlags pkg [
|
||||
"-O3"
|
||||
"-march=znver2"
|
||||
"-mtune=znver2"
|
||||
];
|
||||
};
|
||||
modules =
|
||||
[
|
||||
|
||||
@ -3,7 +3,7 @@
|
||||
service_configs,
|
||||
config,
|
||||
inputs,
|
||||
optimizeWithFlags,
|
||||
optimizePackage,
|
||||
...
|
||||
}:
|
||||
{
|
||||
@ -22,17 +22,13 @@
|
||||
port = service_configs.ports.llama_cpp;
|
||||
host = "0.0.0.0";
|
||||
package = (
|
||||
optimizeWithFlags
|
||||
(inputs.llamacpp.packages.${pkgs.system}.default.overrideAttrs (old: {
|
||||
optimizePackage (
|
||||
inputs.llamacpp.packages.${pkgs.system}.default.overrideAttrs (old: {
|
||||
cmakeFlags = old.cmakeFlags ++ [
|
||||
"-DGGML_AVX2=ON"
|
||||
];
|
||||
}))
|
||||
[
|
||||
"-O3"
|
||||
"-march=znver2"
|
||||
"-mtune=znver2"
|
||||
]
|
||||
})
|
||||
)
|
||||
);
|
||||
extraFlags = [
|
||||
"--flash-attn"
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user