diff options
author | viric <viric@viric.name> | 2016-11-11 10:58:54 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2016-11-11 10:58:54 +0100 |
commit | 5de8637c4efdbb31cc9f78bc9d645578f34d9d69 (patch) | |
tree | 45c2e4b117a521f3595ec51093533220c03a9826 /pkgs/development/libraries | |
parent | ed56263e5d852c3365c998490ff5e1febe3455e6 (diff) | |
parent | 30c1ad5ca378ff46bc90f37bf04e14f9765d6f23 (diff) | |
download | nixlib-5de8637c4efdbb31cc9f78bc9d645578f34d9d69.tar nixlib-5de8637c4efdbb31cc9f78bc9d645578f34d9d69.tar.gz nixlib-5de8637c4efdbb31cc9f78bc9d645578f34d9d69.tar.bz2 nixlib-5de8637c4efdbb31cc9f78bc9d645578f34d9d69.tar.lz nixlib-5de8637c4efdbb31cc9f78bc9d645578f34d9d69.tar.xz nixlib-5de8637c4efdbb31cc9f78bc9d645578f34d9d69.tar.zst nixlib-5de8637c4efdbb31cc9f78bc9d645578f34d9d69.zip |
Merge pull request #20154 from mdaiter/opencv3_extras
opencv3: enable cuda compilations
Diffstat (limited to 'pkgs/development/libraries')
-rw-r--r-- | pkgs/development/libraries/opencv/3.x.nix | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/pkgs/development/libraries/opencv/3.x.nix b/pkgs/development/libraries/opencv/3.x.nix index 0167011ed49a..701aa8e26b4a 100644 --- a/pkgs/development/libraries/opencv/3.x.nix +++ b/pkgs/development/libraries/opencv/3.x.nix @@ -14,6 +14,7 @@ , enableFfmpeg ? false, ffmpeg , enableGStreamer ? false, gst_all_1 , enableEigen ? false, eigen +, enableCuda ? false, cudatoolkit, gcc49 }: let @@ -74,6 +75,7 @@ stdenv.mkDerivation rec { ++ lib.optional enableFfmpeg ffmpeg ++ lib.optionals enableGStreamer (with gst_all_1; [ gstreamer gst-plugins-base ]) ++ lib.optional enableEigen eigen + ++ lib.optional enableCuda [ cudatoolkit gcc49 ] ; propagatedBuildInputs = lib.optional enablePython pythonPackages.numpy; @@ -90,7 +92,10 @@ stdenv.mkDerivation rec { (opencvFlag "JPEG" enableJPEG) (opencvFlag "PNG" enablePNG) (opencvFlag "OPENEXR" enableEXR) - ] ++ lib.optionals enableContrib [ "-DOPENCV_EXTRA_MODULES_PATH=${contribSrc}/modules" ]; + (opencvFlag "CUDA" enableCuda) + (opencvFlag "CUBLAS" enableCuda) + ] ++ lib.optionals enableContrib [ "-DOPENCV_EXTRA_MODULES_PATH=${contribSrc}/modules" ] + ++ lib.optionals enableCuda [ "-DCUDA_FAST_MATH=ON" ]; enableParallelBuilding = true; |