depot/third_party/nixpkgs/pkgs/development/python-modules/paddle_bfloat/default.nix
Default email 9c6ee729d6 Project import generated by Copybara.
GitOrigin-RevId: 6cee3b5893090b0f5f0a06b4cf42ca4e60e5d222
2023-07-15 19:15:38 +02:00

37 lines
931 B
Nix

{ lib
, buildPythonPackage
, fetchPypi
, pythonOlder
, pythonAtLeast
, python
}:
let
pname = "paddle_bfloat";
version = "0.1.7";
format = "wheel";
pyShortVersion = "cp${builtins.replaceStrings ["."] [""] python.pythonVersion}";
binary-hash = (import ./binary-hashes.nix)."${pyShortVersion}" or {};
src = fetchPypi ({
inherit pname version format;
dist = pyShortVersion;
python = pyShortVersion;
abi = pyShortVersion;
platform = "manylinux_2_17_x86_64.manylinux2014_x86_64";
} // binary-hash);
in
buildPythonPackage {
inherit pname version format src;
disabled = pythonOlder "3.9" || pythonAtLeast "3.12";
propagatedBuildInputs = [
];
meta = with lib; {
description = "Paddle numpy bfloat16 package";
homepage = "https://pypi.org/project/paddle-bfloat";
license = licenses.asl20;
maintainers = with maintainers; [ happysalada ];
platforms = platforms.linux;
};
}