pkgs/development/python-modules/reikna/default.nix

   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
  14 buildPythonPackage 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 }