diff options
author | Reno Dakota <170618376+paparodeo@users.noreply.github.com> | 2024-06-17 03:05:58 +0000 |
---|---|---|
committer | Reno Dakota <170618376+paparodeo@users.noreply.github.com> | 2024-06-17 04:58:43 +0000 |
commit | d2b892c753ec8ef2bb7d491f0453a6ef517961f1 (patch) | |
tree | 69d840cb1ec6575d226f044affc06cb7c6909640 /pkgs/development/libraries | |
parent | 80f3f19e1edb6c373c51764779544f89cd24ca73 (diff) |
faiss: refactor buildPhase, installPhase and fixupPhase
use buildFlags to specify the targets and eliminate the need for custom phases. Move remaining code from {build,install,fixup}Phase to post{Build,Install,Fixup)
Diffstat (limited to 'pkgs/development/libraries')
-rw-r--r-- | pkgs/development/libraries/science/math/faiss/default.nix | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/pkgs/development/libraries/science/math/faiss/default.nix b/pkgs/development/libraries/science/math/faiss/default.nix index aedb240caadd7..59494fadfd900 100644 --- a/pkgs/development/libraries/science/math/faiss/default.nix +++ b/pkgs/development/libraries/science/math/faiss/default.nix @@ -97,20 +97,21 @@ stdenv.mkDerivation { "-DCUDAToolkit_INCLUDE_DIR=${cudaJoined}/include" ]; + buildFlags = [ + "faiss" + "demo_ivfpq_indexing" + ] ++ lib.optionals pythonSupport [ + "swigfaiss" + ]; # pip wheel->pip install commands copied over from opencv4 - buildPhase = '' - make -j faiss - make demo_ivfpq_indexing - '' + lib.optionalString pythonSupport '' - make -j swigfaiss + postBuild = lib.optionalString pythonSupport '' (cd faiss/python && python -m pip wheel --verbose --no-index --no-deps --no-clean --no-build-isolation --wheel-dir dist .) ''; - installPhase = '' - make install + postInstall = '' mkdir -p $demos/bin cp ./demos/demo_ivfpq_indexing $demos/bin/ '' + lib.optionalString pythonSupport '' @@ -118,7 +119,7 @@ stdenv.mkDerivation { (cd faiss/python && python -m pip install dist/*.whl --no-index --no-warn-script-location --prefix="$out" --no-cache) ''; - fixupPhase = lib.optionalString (pythonSupport && cudaSupport) '' + postFixup = lib.optionalString (pythonSupport && cudaSupport) '' addOpenGLRunpath $out/${pythonPackages.python.sitePackages}/faiss/*.so addOpenGLRunpath $demos/bin/* ''; |