pyrox.dev
1{ lib
2, fetchPypi
3, buildPythonPackage
4, sphinx
5, pytest-cov
6, pytest
7, Mako
8, numpy
9, funcsigs
10, withCuda ? false, pycuda
11, withOpenCL ? true, pyopencl
12}:
13
14buildPythonPackage rec {
15 pname = "reikna";
16 version = "0.7.6";
17
18 src = fetchPypi {
19 inherit pname version;
20 sha256 = "722fefbd253d0bbcbf5250b7b9c4aca5722cde4ca38bfbf863a551a5fc26edfa";
21 };
22
23 checkInputs = [ sphinx pytest-cov pytest ];
24
25 propagatedBuildInputs = [ Mako numpy funcsigs ]
26 ++ lib.optional withCuda pycuda
27 ++ lib.optional withOpenCL pyopencl;
28
29 checkPhase = ''
30 py.test
31 '';
32
33 # Requires device
34 doCheck = false;
35
36 meta = with lib; {
37 description = "GPGPU algorithms for PyCUDA and PyOpenCL";
38 homepage = "https://github.com/fjarri/reikna";
39 license = licenses.mit;
40 maintainers = [ maintainers.fridh ];
41
42 };
43
44}