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

   1 {
   2   lib,
   3   stdenv,
   4   fetchFromGitHub,
   5   buildPythonPackage,
   6
   7   # build-system
   8   cmake,
   9   nanobind,
  10   ninja,
  11   numpy,
  12   scikit-build-core,
  13
  14   # buildInputs
  15   opencl-headers,
  16   pybind11,
  17   darwin,
  18   ocl-icd,
  19
  20   # dependencies
  21   platformdirs,
  22   pytools,
  23
  24   # tests
  25   pytestCheckHook,
  26 }:
  27
  28 let
  29   os-specific-buildInputs =
  30     if stdenv.hostPlatform.isDarwin then [ darwin.apple_sdk.frameworks.OpenCL ] else [ ocl-icd ];
  31 in
  32 buildPythonPackage rec {
  33   pname = "pyopencl";
  34   version = "2024.3";
  35   pyproject = true;
  36
  37   src = fetchFromGitHub {
  38     owner = "inducer";
  39     repo = "pyopencl";
  40     rev = "refs/tags/v${version}";
  41     fetchSubmodules = true;
  42     hash = "sha256-HE7dARgKnZxqjAXX4iI1ml0N2BalyTo+ZAzjC2ThEN8=";
  43   };
  44
  45   build-system = [
  46     cmake
  47     nanobind
  48     ninja
  49     numpy
  50     scikit-build-core
  51   ];
  52
  53   dontUseCmakeConfigure = true;
  54
  55   buildInputs = [
  56     opencl-headers
  57     pybind11
  58   ] ++ os-specific-buildInputs;
  59
  60   dependencies = [
  61     numpy
  62     platformdirs
  63     pytools
  64   ];
  65
  66   nativeCheckInputs = [ pytestCheckHook ];
  67
  68   preCheck = ''
  69     export HOME=$(mktemp -d)
  70
  71     # https://github.com/NixOS/nixpkgs/issues/255262
  72     cd $out
  73   '';
  74
  75   # https://github.com/inducer/pyopencl/issues/784 Note that these failing
  76   # tests are all the tests that are available.
  77   doCheck = false;
  78
  79   pythonImportsCheck = [
  80     "pyopencl"
  81     "pyopencl.array"
  82     "pyopencl.cltypes"
  83     "pyopencl.compyte"
  84     "pyopencl.elementwise"
  85     "pyopencl.tools"
  86   ];
  87
  88   meta = {
  89     description = "Python wrapper for OpenCL";
  90     homepage = "https://github.com/pyopencl/pyopencl";
  91     changelog = "https://github.com/inducer/pyopencl/releases/tag/v${version}";
  92     license = lib.licenses.mit;
  93     maintainers = with lib.maintainers; [ GaetanLepage ];
  94     # ld: symbol(s) not found for architecture arm64
  95     broken = stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isAarch64;
  96   };
  97 }