@@ -1,4 +1,4 @@
1−{ cmake, fetchFromGitHub, opencv3, stdenv, ocl-icd, opencl-headers
1+{ cmake, fetchFromGitHub, makeWrapper, opencv3, stdenv, ocl-icd, opencl-headers
22, cudaSupport ? false, cudatoolkit ? null
33}:
44
@@ -19,10 +19,14 @@ stdenv.mkDerivation rec {
1919 '';
2020
2121 buildInputs = [
22− opencv3 opencl-headers ocl-icd
22+ ocl-icd opencv3 opencl-headers
2323 ] ++ stdenv.lib.optional cudaSupport cudatoolkit;
2424
25− nativeBuildInputs = [ cmake ];
25+ nativeBuildInputs = [ cmake makeWrapper ];
26+
27+ preFixup = ''
28+ wrapProgram $out/bin/waifu2x-converter-cpp --prefix LD_LIBRARY_PATH : "${ocl-icd}/lib"
29+ '';
2630
2731 meta = {
2832 description = "Improved fork of Waifu2X C++ using OpenCL and OpenCV";