Unverified Commit 5fbdef57 authored by illustris's avatar illustris
Browse files

python3Packages.lightgbm: add GPU support

parent c1329a14

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

+18 −5
Original line number Diff line number Diff line
@@ -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 ];

  };

}