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
92
93
94
95
96
97
98
99
|
{
lib,
stdenv,
fetchFromGitHub,
buildPythonPackage,
# build-system
cmake,
nanobind,
ninja,
numpy,
scikit-build-core,
# buildInputs
opencl-headers,
pybind11,
darwin,
ocl-icd,
# dependencies
platformdirs,
pytools,
# tests
pytestCheckHook,
writableTmpDirAsHomeHook,
mako,
pocl,
}:
buildPythonPackage rec {
pname = "pyopencl";
version = "2025.1";
pyproject = true;
src = fetchFromGitHub {
owner = "inducer";
repo = "pyopencl";
tag = "v${version}";
fetchSubmodules = true;
hash = "sha256-wAZBDPMJbTmujP1j7LjK28ZozZaUwKPDPZLZbFFTeAs=";
};
build-system = [
cmake
nanobind
ninja
numpy
scikit-build-core
];
dontUseCmakeConfigure = true;
buildInputs = [
opencl-headers
ocl-icd
pybind11
];
dependencies = [
numpy
platformdirs
pytools
];
nativeCheckInputs = [
pocl
mako
pytestCheckHook
writableTmpDirAsHomeHook
] ++ pytools.optional-dependencies.siphash;
env = {
CL_INC_DIR = "${opencl-headers}/include";
CL_LIB_DIR = "${ocl-icd}/lib";
CL_LIBNAME = "${ocl-icd}/lib/libOpenCL${stdenv.hostPlatform.extensions.sharedLibrary}";
};
preCheck = ''
rm -rf pyopencl
'';
pythonImportsCheck = [
"pyopencl"
"pyopencl.array"
"pyopencl.cltypes"
"pyopencl.compyte"
"pyopencl.elementwise"
"pyopencl.tools"
];
meta = {
description = "Python wrapper for OpenCL";
homepage = "https://github.com/pyopencl/pyopencl";
changelog = "https://github.com/inducer/pyopencl/releases/tag/v${version}";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ GaetanLepage ];
};
}
|
