56 lines
1.0 KiB
Nix
56 lines
1.0 KiB
Nix
{ lib
|
|
, stdenv
|
|
, fetchFromGitHub
|
|
, cmake
|
|
, vapoursynth
|
|
, cudatoolkit
|
|
, cudaPackages
|
|
}:
|
|
|
|
stdenv.mkDerivation rec {
|
|
pname = "vstrt";
|
|
version = "10";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "AmusementClub";
|
|
repo = "vs-mlrt";
|
|
rev = "v${version}";
|
|
sha256 = "sha256-fnnVaWPvT/eqr1WEwyhX8zi26IK3Ndm7UHLkSLI1nFk=";
|
|
};
|
|
|
|
sourceRoot = "source/vstrt";
|
|
|
|
patches = [
|
|
./no-git-call-in-cmake.patch
|
|
];
|
|
|
|
nativeBuildInputs = [
|
|
cmake
|
|
];
|
|
|
|
buildInputs = [
|
|
vapoursynth
|
|
cudatoolkit
|
|
cudaPackages.tensorrt
|
|
];
|
|
|
|
cmakeFlags = [
|
|
"-DVCS_TAG=v${version}"
|
|
"-DCMAKE_CXX_FLAGS=-I${vapoursynth}/include/vapoursynth"
|
|
"-DCMAKE_SKIP_RPATH=ON"
|
|
];
|
|
|
|
postInstall = ''
|
|
mkdir $out/lib/vapoursynth
|
|
ln -s $out/lib/libvstrt.so $out/lib/vapoursynth
|
|
'';
|
|
|
|
meta = with lib; {
|
|
description = "TensorRT-based GPU Runtime";
|
|
homepage = "https://github.com/AmusementClub/vs-mlrt";
|
|
license = licenses.gpl3;
|
|
maintainers = with maintainers; [ aidalgol ];
|
|
platforms = platforms.all;
|
|
};
|
|
}
|