Commit 8b2b3b6a authored by Someone Serge's avatar Someone Serge
Browse files

faiss: use autoAddDriverRunpath

parent 4076652b
Loading
Loading
Loading
Loading
+2 −7
Original line number Diff line number Diff line
@@ -11,7 +11,7 @@
  llvmPackages,
  blas,
  swig,
  addDriverRunpath,
  autoAddDriverRunpath,
  optLevel ?
    let
      optLevels =
@@ -72,7 +72,7 @@ stdenv.mkDerivation {
    [ cmake ]
    ++ lib.optionals cudaSupport [
      cudaPackages.cuda_nvcc
      addDriverRunpath
      autoAddDriverRunpath
    ]
    ++ lib.optionals pythonSupport [
      pythonPackages.python
@@ -127,11 +127,6 @@ stdenv.mkDerivation {
      cp faiss/python/dist/*.whl "$dist/"
    '';

  postFixup = lib.optionalString (pythonSupport && cudaSupport) ''
    addDriverRunpath $out/${pythonPackages.python.sitePackages}/faiss/*.so
    addDriverRunpath $demos/bin/*
  '';

  passthru = {
    inherit cudaSupport cudaPackages pythonSupport;