nixpkgs/pkgs/development/libraries/ncnn/default.nix
2022-02-25 11:27:16 -08:00

51 lines
1.1 KiB
Nix

{ lib
, stdenv
, fetchFromGitHub
, cmake
, vulkan-headers
, vulkan-loader
, glslang
, opencv
, protobuf
}:
stdenv.mkDerivation rec {
pname = "ncnn";
version = "20220216";
src = fetchFromGitHub {
owner = "Tencent";
repo = pname;
rev = version;
sha256 = "sha256-QHLD5NQZA7WR4mRQ0NIaXuAu59IV4SjXHOOlar5aOew=";
};
patches = [
./cmakelists.patch
./gpu-include.patch
];
cmakeFlags = [
"-DNCNN_CMAKE_VERBOSE=1" # Only for debugging the build
"-DNCNN_SHARED_LIB=1"
"-DNCNN_ENABLE_LTO=1"
"-DNCNN_VULKAN=1"
"-DNCNN_BUILD_EXAMPLES=0"
"-DNCNN_BUILD_TOOLS=0"
"-DNCNN_SYSTEM_GLSLANG=1"
"-DNCNN_PYTHON=0" # Should be an attribute
"-DGLSLANG_TARGET_DIR=${glslang}/lib/cmake"
];
nativeBuildInputs = [ cmake ];
buildInputs = [ vulkan-headers vulkan-loader glslang opencv protobuf ];
meta = with lib; {
description = "ncnn is a high-performance neural network inference framework optimized for the mobile platform";
homepage = "https://github.com/Tencent/ncnn";
license = licenses.bsd3;
maintainers = with maintainers; [ tilcreator ];
};
}