pkgs/development/python-modules/paddle-bfloat/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , pythonOlder
   5 , pythonAtLeast
   6 , numpy
   7 }:
   8 let
   9   pname = "paddle-bfloat";
  10   version = "0.1.7";
  11 in
  12 buildPythonPackage {
  13   inherit pname version;
  14   format = "setuptools";
  15
  16   src = fetchPypi {
  17     pname = "paddle_bfloat";
  18     inherit version;
  19     hash = "sha256-mrjQCtLsXOvqeHHMjuMx65FvMfZ2+wTh1ao9ZJE+9xw=";
  20   };
  21
  22   postPatch = ''
  23     sed '1i#include <memory>' -i bfloat16.cc # gcc12
  24     # replace deprecated function for python3.11
  25     substituteInPlace bfloat16.cc \
  26       --replace "Py_TYPE(&NPyBfloat16_Descr) = &PyArrayDescr_Type" "Py_SET_TYPE(&NPyBfloat16_Descr, &PyArrayDescr_Type)"
  27   '';
  28
  29   disabled = pythonOlder "3.9" || pythonAtLeast "3.12";
  30
  31   propagatedBuildInputs = [
  32     numpy
  33   ];
  34
  35   pythonImportsCheck = [ "paddle_bfloat" ];
  36
  37 # upstream has no tests
  38   doCheck = false;
  39
  40   meta = with lib; {
  41     description = "Paddle numpy bfloat16 package";
  42     homepage = "https://pypi.org/project/paddle-bfloat";
  43     license = licenses.asl20;
  44     maintainers = with maintainers; [ happysalada ];
  45   };
  46 }