b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Merge pull request #221775 from illustris/lightgbm-gpu

Browse source
This commit is contained in:
Sandro 2023-04-15 00:03:39 +02:00 committed by GitHub
parent 944a4d5aa0 5fbdef5720
commit b75e0f61cf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
pkgs/development/python-modules/lightgbm/default.nix
View file

@ -8,6 +8,11 @@
, scikit-learn
, llvmPackages ? null
, pythonOlder
, python
, ocl-icd
, opencl-headers
, boost
, gpuSupport ? true
}:
buildPythonPackage rec {
@ -28,9 +33,13 @@ buildPythonPackage rec {
dontUseCmakeConfigure = true;
buildInputs = lib.optionals stdenv.cc.isClang [
buildInputs = (lib.optionals stdenv.cc.isClang [
llvmPackages.openmp
];
]) ++ (lib.optionals gpuSupport [
boost
ocl-icd
opencl-headers
]);
propagatedBuildInputs = [
numpy
@ -38,6 +47,10 @@ buildPythonPackage rec {
scikit-learn
];
buildPhase = ''
${python.pythonForBuild.interpreter} setup.py bdist_wheel ${lib.optionalString gpuSupport "--gpu"}
'';
postConfigure = ''
export HOME=$(mktemp -d)
'';
@ -51,11 +64,11 @@ buildPythonPackage rec {
"lightgbm"
];
meta = with lib; {
meta = {
description = "A fast, distributed, high performance gradient boosting (GBDT, GBRT, GBM or MART) framework";
homepage = "https://github.com/Microsoft/LightGBM";
changelog = "https://github.com/microsoft/LightGBM/releases/tag/v${version}";
license = licenses.mit;
maintainers = with maintainers; [ teh costrouc ];
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ teh costrouc ];
};
}