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 , cudaSupport ? false, cudatoolkit ? null
}: }:
@ -19,10 +19,14 @@ stdenv.mkDerivation rec {
''; '';
buildInputs = [ buildInputs = [
opencv3 opencl-headers ocl-icd ocl-icd opencv3 opencl-headers
] ++ stdenv.lib.optional cudaSupport cudatoolkit; ] ++ 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 = { meta = {
description = "Improved fork of Waifu2X C++ using OpenCL and OpenCV"; description = "Improved fork of Waifu2X C++ using OpenCL and OpenCV";