waifu2x-converter-cpp: add OpenCL support

This commit is contained in:
Simon Bruder 2020-10-17 20:05:13 +02:00
parent c69dd49995
commit ca3dd4b52f
No known key found for this signature in database
GPG key ID: 6F03E0000CC5B62F

View file

@ -1,4 +1,4 @@
{ cmake, fetchFromGitHub, opencv3, stdenv, ocl-icd, opencl-headers
{ cmake, fetchFromGitHub, makeWrapper, opencv3, stdenv, ocl-icd, opencl-headers
, cudaSupport ? false, cudatoolkit ? null
}:
@ -19,10 +19,14 @@ stdenv.mkDerivation rec {
'';
buildInputs = [
opencv3 opencl-headers ocl-icd
ocl-icd opencv3 opencl-headers
] ++ stdenv.lib.optional cudaSupport cudatoolkit;
nativeBuildInputs = [ cmake ];
nativeBuildInputs = [ cmake makeWrapper ];
preFixup = ''
wrapProgram $out/bin/waifu2x-converter-cpp --prefix LD_LIBRARY_PATH : "${ocl-icd}/lib"
'';
meta = {
description = "Improved fork of Waifu2X C++ using OpenCL and OpenCV";