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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
{
lib,
stdenv,
buildPythonPackage,
fetchFromGitHub,
# build-system
cmake,
ninja,
pathspec,
scikit-build-core,
# dependencies
eigen,
# tests
pytestCheckHook,
numpy,
scipy,
torch,
tensorflow-bin,
jax,
jaxlib,
nanobind,
}:
buildPythonPackage rec {
pname = "nanobind";
version = "2.10.2";
pyproject = true;
src = fetchFromGitHub {
owner = "wjakob";
repo = "nanobind";
tag = "v${version}";
fetchSubmodules = true;
hash = "sha256-io44YhN+VpfHFWyvvLWSanRgbzA0whK8WlDNRi3hahU=";
};
build-system = [
cmake
ninja
pathspec
scikit-build-core
];
dependencies = [ eigen ];
dontUseCmakeBuildDir = true;
# nanobind check requires heavy dependencies such as tensorflow
# which are less than ideal to be imported in children packages that
# use it as build-system parameter.
doCheck = false;
preCheck = ''
# build tests
make -j $NIX_BUILD_CORES
'';
nativeCheckInputs = [
pytestCheckHook
numpy
scipy
torch
]
++ lib.optionals (lib.meta.availableOn stdenv.hostPlatform tensorflow-bin) [
tensorflow-bin
jax
jaxlib
];
passthru.tests = {
pytest = nanobind.overridePythonAttrs { doCheck = true; };
};
meta = {
homepage = "https://github.com/wjakob/nanobind";
changelog = "https://github.com/wjakob/nanobind/blob/${src.tag}/docs/changelog.rst";
description = "Tiny and efficient C++/Python bindings";
longDescription = ''
nanobind is a small binding library that exposes C++ types in Python and
vice versa. It is reminiscent of Boost.Python and pybind11 and uses
near-identical syntax. In contrast to these existing tools, nanobind is
more efficient: bindings compile in a shorter amount of time, produce
smaller binaries, and have better runtime performance.
'';
license = lib.licenses.bsd3;
maintainers = with lib.maintainers; [ parras ];
};
}
|
