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

   1 { lib
   2 , stdenv
   3 , fetchPypi
   4 , buildPythonPackage
   5 , appdirs
   6 , cffi
   7 , decorator
   8 , mako
   9 , mesa_drivers
  10 , numpy
  11 , ocl-icd
  12 , oldest-supported-numpy
  13 , opencl-headers
  14 , platformdirs
  15 , pybind11
  16 , pytest
  17 , pytestCheckHook
  18 , pytools
  19 , setuptools
  20 , six
  21 , wheel
  22 }:
  23
  24 let
  25   os-specific-buildInputs =
  26     if stdenv.isDarwin then [ mesa_drivers.dev ] else [ ocl-icd ];
  27 in buildPythonPackage rec {
  28   pname = "pyopencl";
  29   version = "2023.1.4";
  30   format = "pyproject";
  31
  32   src = fetchPypi {
  33     inherit pname version;
  34     hash = "sha256-IgF078qQDp1d5a7yqht3pvJVBQHekrA1qRATrq5NTF4=";
  35   };
  36
  37   nativeBuildInputs = [
  38     oldest-supported-numpy
  39     setuptools
  40     wheel
  41   ];
  42
  43   buildInputs = [ opencl-headers pybind11 ] ++ os-specific-buildInputs;
  44
  45   propagatedBuildInputs = [
  46     appdirs
  47     cffi
  48     decorator
  49     mako
  50     numpy
  51     platformdirs
  52     pytools
  53     six
  54   ];
  55
  56   nativeCheckInputs = [ pytestCheckHook ];
  57
  58   preBuild = ''
  59     export HOME=$(mktemp -d)
  60   '';
  61
  62   # gcc: error: pygpu_language_opencl.cpp: No such file or directory
  63   doCheck = false;
  64
  65   pythonImportsCheck = [ "pyopencl" ];
  66
  67   meta = with lib; {
  68     description = "Python wrapper for OpenCL";
  69     homepage = "https://github.com/pyopencl/pyopencl";
  70     license = licenses.mit;
  71     maintainers = [ maintainers.fridh ];
  72   };
  73 }