blob: f26d52a0b6fdda009c89803f02b6c939badc0c83 (
plain)
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
51
52
53
54
55
56
|
{
lib,
stdenv,
fetchFromGitHub,
cmake,
vulkan-headers,
vulkan-loader,
glslang,
opencv,
protobuf,
}:
stdenv.mkDerivation (finalAttrs: {
pname = "ncnn";
version = "20250503";
src = fetchFromGitHub {
owner = "Tencent";
repo = "ncnn";
tag = finalAttrs.version;
hash = "sha256-7wktoeei16QaPdcxVVS25sZYPhTQMEq9PjaHBwm5Eas=";
};
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.hostPlatform.isDarwin [ "-DVulkan_LIBRARY=-lvulkan" ];
nativeBuildInputs = [ cmake ];
buildInputs = [
vulkan-headers
vulkan-loader
glslang
opencv
protobuf
];
meta = {
description = "Neural network inference framework";
homepage = "https://github.com/Tencent/ncnn";
license = lib.licenses.bsd3;
maintainers = with lib.maintainers; [ tilcreator ];
platforms = lib.platforms.all;
};
})
|
