blob: 8c0f213b82980cbf51e50c9dec0db45fa6773a5f (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
{ 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 ];
};
}
|