blob: ae82d0490be4b0830f0205a98c7c2f050361dcaf (
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 = "20240820";
src = fetchFromGitHub {
owner = "Tencent";
repo = pname;
rev = version;
hash = "sha256-KFRWpPajSqYeasPKaNMVe0WTIXwCI5v9GLo5ygN/22M=";
};
patches = [
./cmakelists.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
]
# Requires setting `Vulkan_LIBRARY` on Darwin. Otherwise the build fails due to missing symbols.
++ lib.optionals stdenv.isDarwin [ "-DVulkan_LIBRARY=-lvulkan" ];
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 ];
platforms = platforms.all;
};
}
|
