turingdb: fix cudaSupport

This commit is contained in:
Gaetan Lepage
2026-06-14 19:02:38 +00:00
parent bcb2907c62
commit 81707a38fa
+22 -2
View File
@@ -21,15 +21,27 @@
zlib,
llvmPackages_20,
versionCheckHook,
config,
cudaSupport ? config.cudaSupport,
cudaPackages,
}:
let
turingstdenv = if stdenv.hostPlatform.isDarwin then llvmPackages_20.stdenv else stdenv;
turingstdenv =
if stdenv.hostPlatform.isDarwin then
llvmPackages_20.stdenv
else if cudaSupport then
cudaPackages.backendStdenv
else
stdenv;
in
turingstdenv.mkDerivation (finalAttrs: {
pname = "turingdb";
version = "1.33";
__structuredAttrs = true;
src = fetchFromGitHub {
owner = "turing-db";
repo = "turingdb";
@@ -74,6 +86,10 @@ turingstdenv.mkDerivation (finalAttrs: {
gitMinimal
pkg-config
python3
]
++ lib.optionals cudaSupport [
# Needed by transitive dependency faiss
cudaPackages.cuda_nvcc
];
buildInputs = [
@@ -90,7 +106,11 @@ turingstdenv.mkDerivation (finalAttrs: {
zlib
]
++ lib.optionals turingstdenv.isDarwin [ llvmPackages_20.openmp ]
++ lib.optionals stdenv.hostPlatform.isLinux [ stdenv.cc.cc.lib ];
++ lib.optionals stdenv.hostPlatform.isLinux [ stdenv.cc.cc.lib ]
++ lib.optionals cudaSupport [
cudaPackages.cuda_cudart
cudaPackages.libcublas
];
cmakeFlags = [
(lib.cmakeBool "NIX_BUILD" true)