summary refs log tree commit diff
path: root/pkgs/development/libraries
diff options
context:
space:
mode:
authorviric <viric@viric.name>2016-11-11 10:58:54 +0100
committerGitHub <noreply@github.com>2016-11-11 10:58:54 +0100
commit5de8637c4efdbb31cc9f78bc9d645578f34d9d69 (patch)
tree45c2e4b117a521f3595ec51093533220c03a9826 /pkgs/development/libraries
parented56263e5d852c3365c998490ff5e1febe3455e6 (diff)
parent30c1ad5ca378ff46bc90f37bf04e14f9765d6f23 (diff)
downloadnixlib-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.nix7
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;