ilyakooo0/nixpkgs
1
0
Fork 0
mirror of https://github.com/ilyakooo0/nixpkgs.git synced 2024-09-20 20:20:26 +03:00
Code Issues Projects Releases Wiki Activity

sqlite-vss: init at 0.1.2

Browse Source
This commit is contained in:
Enno Richter 2023-10-22 14:54:37 +02:00
parent f9c7c12de6
commit ebafcea766
2 changed files with 94 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
pkgs/by-name/sq/sqlite-vss/package.nix Normal file
View File

@ -0,0 +1,53 @@
{ lib
, cmake
, faiss
, fetchFromGitHub
, gomp
, llvmPackages
, nlohmann_json
, sqlite
, stdenv
}:
stdenv.mkDerivation (finalAttrs: {
pname = "sqlite-vss";
version = "0.1.2";
src = fetchFromGitHub {
owner = "asg017";
repo = "sqlite-vss";
rev = "v${finalAttrs.version}";
hash = "sha256-cb9UlSUAZp8B5NpNDBvJ2+ung98gjVKLxrM2Ek9fOcs=";
};
patches = [ ./use-nixpkgs-libs.patch ];
nativeBuildInputs = [ cmake ];
buildInputs = [ nlohmann_json faiss sqlite ]
++ lib.optional stdenv.isLinux gomp
++ lib.optional stdenv.isDarwin llvmPackages.openmp;
SQLITE_VSS_CMAKE_VERSION = finalAttrs.version;
installPhase = ''
runHook preInstall
install -Dm444 -t "$out/lib" \
"libsqlite_vector0${stdenv.hostPlatform.extensions.staticLibrary}" \
"libsqlite_vss0${stdenv.hostPlatform.extensions.staticLibrary}" \
"vector0${stdenv.hostPlatform.extensions.sharedLibrary}" \
"vss0${stdenv.hostPlatform.extensions.sharedLibrary}"
runHook postInstall
'';
meta = with lib;{
description = "SQLite extension for efficient vector search based on Faiss";
homepage = "https://github.com/asg017/sqlite-vss";
changelog = "https://github.com/asg017/sqlite-vss/releases/tag/v${finalAttrs.version}";
license = licenses.mit;
maintainers = with maintainers; [ elohmeier ];
platforms = platforms.unix;
};
})

41
pkgs/by-name/sq/sqlite-vss/use-nixpkgs-libs.patch Normal file
View File

@ -0,0 +1,41 @@
diff --git a/CMakeLists.txt b/CMakeLists.txt
index c59d993..5606b46 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -18,15 +18,12 @@ set(CMAKE_CXX_STANDARD_REQUIRED ON)
option(FAISS_ENABLE_GPU "" OFF)
option(FAISS_ENABLE_PYTHON "" OFF)
option(BUILD_TESTING "" OFF)
-add_subdirectory(./vendor/faiss)
-
-# vendor in SQLite amalgammation
-include_directories(vendor/sqlite)
-link_directories(BEFORE vendor/sqlite)
+find_package(OpenMP REQUIRED)
+find_package(faiss REQUIRED)
# Adding nlohmann_json for json parsing
set(JSON_BuildTests OFF CACHE INTERNAL "")
-add_subdirectory(vendor/json)
+find_package(nlohmann_json REQUIRED)
# ================================== sqlite-vector ================================== #
add_library(sqlite-vector SHARED src/sqlite-vector.cpp)
@@ -49,7 +46,7 @@ target_compile_definitions(sqlite-vector-static PUBLIC SQLITE_CORE)
# ================================== sqlite-vss ================================== #
add_library(sqlite-vss SHARED src/sqlite-vss.cpp)
target_link_libraries(sqlite-vss sqlite3)
-target_link_libraries(sqlite-vss faiss_avx2)
+target_link_libraries(sqlite-vss faiss)
target_include_directories(sqlite-vss PUBLIC "${PROJECT_BINARY_DIR}")
set_target_properties(sqlite-vss PROPERTIES PREFIX "")
@@ -58,7 +55,7 @@ set_target_properties(sqlite-vss PROPERTIES OUTPUT_NAME "vss0")
# ============================== sqlite-vss-static =============================== #
add_library(sqlite-vss-static STATIC src/sqlite-vss.cpp)
target_link_libraries(sqlite-vss-static PRIVATE sqlite3)
-target_link_libraries(sqlite-vss-static PUBLIC faiss_avx2)
+target_link_libraries(sqlite-vss-static PUBLIC faiss)
target_link_options(sqlite-vss-static PRIVATE "-Wl,-all_load")
target_include_directories(sqlite-vss-static PUBLIC "${PROJECT_BINARY_DIR}")
set_target_properties(sqlite-vss-static PROPERTIES OUTPUT_NAME "sqlite_vss0")