nixpkgs/pkgs/development/python-modules/reikna/default.nix
2021-07-21 13:32:17 +02:00

45 lines
822 B
Nix

{ lib
, fetchPypi
, buildPythonPackage
, sphinx
, pytest-cov
, pytest
, Mako
, numpy
, funcsigs
, withCuda ? false, pycuda
, withOpenCL ? true, pyopencl
}:
buildPythonPackage rec {
pname = "reikna";
version = "0.7.5";
src = fetchPypi {
inherit pname version;
sha256 = "d01f4264c8379ef2962a93aacb002d491b92ef9b5b22b45f77e7821dfa87bef7";
};
checkInputs = [ sphinx pytest-cov pytest ];
propagatedBuildInputs = [ Mako numpy funcsigs ]
++ lib.optional withCuda pycuda
++ lib.optional withOpenCL pyopencl;
checkPhase = ''
py.test
'';
# Requires device
doCheck = false;
meta = with lib; {
description = "GPGPU algorithms for PyCUDA and PyOpenCL";
homepage = "https://github.com/fjarri/reikna";
license = licenses.mit;
maintainers = [ maintainers.fridh ];
};
}