opencv4: propagate optical flow sdk same as cuda

This commit is contained in:
Someone Serge 2023-11-30 00:59:10 +00:00
parent ada3991349
commit 4569838029
No known key found for this signature in database
GPG Key ID: 7B0E3B1390D61DA4
2 changed files with 7 additions and 2 deletions

View File

@ -18,6 +18,11 @@ stdenv.mkDerivation {
cp -R * $out/include
'';
postFixup = ''
mkdir -p $out/nix-support
echo $pname >> "$out/nix-support/include-in-cudatoolkit-root"
'';
meta = with lib; {
description = "Nvidia optical flow headers for computing the relative motion of pixels between images";
homepage = "https://developer.nvidia.com/opticalflow-sdk";

View File

@ -334,6 +334,7 @@ effectiveStdenv.mkDerivation {
cuda_cudart
cuda_cccl # <thrust/*>
libnpp # npp.h
nvidia-optical-flow-sdk
] ++ lib.optionals enableCublas [
libcublas # cublas_v2.h
] ++ lib.optionals enableCudnn [
@ -342,8 +343,7 @@ effectiveStdenv.mkDerivation {
libcufft # cufft.h
]);
propagatedBuildInputs = lib.optional enablePython pythonPackages.numpy
++ lib.optionals enableCuda [ nvidia-optical-flow-sdk ];
propagatedBuildInputs = lib.optionals enablePython [ pythonPackages.numpy ];
nativeBuildInputs = [ cmake pkg-config unzip ]
++ lib.optionals enablePython [