public/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added cuda cmake flags (allows GPU acceleration of some of the libraries functions)

Browse Source
This commit is contained in:
alexanderWatanabe 2020-04-13 01:22:45 +09:00
parent 35d8514a91
commit c45155d6b7
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pkgs/development/libraries/dlib/default.nix
View File

@ -3,6 +3,7 @@
# see http://dlib.net/compile.html # see http://dlib.net/compile.html
, avxSupport ? true , avxSupport ? true
, cudaSupport ? true
}: }:
stdenv.mkDerivation rec { stdenv.mkDerivation rec {
@ -20,7 +21,9 @@ stdenv.mkDerivation rec {
rm -rf dlib/external rm -rf dlib/external
''; '';
cmakeFlags = [ "-DUSE_AVX_INSTRUCTIONS=${if avxSupport then "yes" else "no"}" ]; cmakeFlags = [
"-DUSE_DLIB_USE_CUDA=${if cudaSupport then "1" else "0"}"
"-DUSE_AVX_INSTRUCTIONS=${if avxSupport then "yes" else "no"}" ];
enableParallelBuilding = true; enableParallelBuilding = true;
nativeBuildInputs = [ cmake pkgconfig ]; nativeBuildInputs = [ cmake pkgconfig ];