about summary refs log tree commit diff
path: root/pkgs/by-name/lo
diff options
context:
space:
mode:
authorChristian Kögler <ck3d@gmx.de>2024-04-07 15:36:57 +0200
committerChristian Kögler <ck3d@gmx.de>2024-04-07 15:36:57 +0200
commit673216d6eb85ffa87f8ffb722c9ee642fa32b315 (patch)
treec63203f5f5aaf5a743713768ec2ef6a7910fb908 /pkgs/by-name/lo
parent45acd0adb5723d0c7523eebebb35b1f30a08aac7 (diff)
local-ai: use defaults from config and stdenv
Diffstat (limited to 'pkgs/by-name/lo')
-rw-r--r--pkgs/by-name/lo/local-ai/package.nix7
1 files changed, 4 insertions, 3 deletions
diff --git a/pkgs/by-name/lo/local-ai/package.nix b/pkgs/by-name/lo/local-ai/package.nix
index 5e282d6380899..373c6df6a10e4 100644
--- a/pkgs/by-name/lo/local-ai/package.nix
+++ b/pkgs/by-name/lo/local-ai/package.nix
@@ -1,4 +1,5 @@
-{ stdenv
+{ config
+, stdenv
 , lib
 , fetchpatch
 , fetchFromGitHub
@@ -21,7 +22,7 @@
   # CPU extensions
 , enable_avx ? true
 , enable_avx2 ? true
-, enable_avx512 ? false
+, enable_avx512 ? stdenv.hostPlatform.avx512Support
 , enable_f16c ? true
 , enable_fma ? true
 
@@ -31,7 +32,7 @@
 , with_openblas ? false
 , openblas
 
-, with_cublas ? false
+, with_cublas ? config.cudaSupport
 , cudaPackages
 
 , with_clblas ? false