xgboost: Enable building on Darwin (#37046)
* xgboost: Enable building on Darwin * Review feedback: pass `xgboost` to the python package
This commit is contained in:
parent
80582c600d
commit
92a77abd31
|
@ -2,6 +2,7 @@
|
|||
, avxSupport ? false
|
||||
, cudaSupport ? false, cudatoolkit
|
||||
, ncclSupport ? false, nccl
|
||||
, llvmPackages
|
||||
}:
|
||||
|
||||
assert ncclSupport -> cudaSupport;
|
||||
|
@ -19,7 +20,7 @@ stdenv.mkDerivation rec {
|
|||
|
||||
enableParallelBuilding = true;
|
||||
|
||||
nativeBuildInputs = [ cmake ];
|
||||
nativeBuildInputs = [ cmake ] ++ lib.optional stdenv.isDarwin llvmPackages.openmp;
|
||||
|
||||
buildInputs = lib.optional cudaSupport cudatoolkit
|
||||
++ lib.optional ncclSupport nccl;
|
||||
|
@ -27,10 +28,12 @@ stdenv.mkDerivation rec {
|
|||
cmakeFlags = lib.optionals cudaSupport [ "-DUSE_CUDA=ON" "-DCUDA_HOST_COMPILER=${cudatoolkit.cc}/bin/cc" ]
|
||||
++ lib.optional ncclSupport "-DUSE_NCCL=ON";
|
||||
|
||||
installPhase = ''
|
||||
installPhase = let
|
||||
libname = if stdenv.isDarwin then "libxgboost.dylib" else "libxgboost.so";
|
||||
in ''
|
||||
mkdir -p $out
|
||||
cp -r ../include $out
|
||||
install -Dm755 ../lib/libxgboost.so $out/lib/libxgboost.so
|
||||
install -Dm755 ../lib/${libname} $out/lib/${libname}
|
||||
install -Dm755 ../xgboost $out/bin/xgboost
|
||||
'';
|
||||
|
||||
|
@ -38,7 +41,7 @@ stdenv.mkDerivation rec {
|
|||
description = "Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library";
|
||||
homepage = https://github.com/dmlc/xgboost;
|
||||
license = licenses.asl20;
|
||||
platforms = [ "x86_64-linux" "i686-linux" ];
|
||||
platforms = [ "x86_64-linux" "i686-linux" "x86_64-darwin" ];
|
||||
maintainers = with maintainers; [ abbradar ];
|
||||
};
|
||||
}
|
||||
|
|
35
pkgs/development/python-modules/xgboost/default.nix
Normal file
35
pkgs/development/python-modules/xgboost/default.nix
Normal file
|
@ -0,0 +1,35 @@
|
|||
{ stdenv
|
||||
, lib
|
||||
, pkgs
|
||||
, buildPythonPackage
|
||||
, nose
|
||||
, scipy
|
||||
, xgboost
|
||||
}:
|
||||
|
||||
buildPythonPackage rec {
|
||||
name = "xgboost-${version}";
|
||||
|
||||
inherit (xgboost) version src meta;
|
||||
|
||||
propagatedBuildInputs = [ scipy ];
|
||||
checkInputs = [ nose ];
|
||||
|
||||
postPatch = let
|
||||
libname = if stdenv.isDarwin then "libxgboost.dylib" else "libxgboost.so";
|
||||
|
||||
in ''
|
||||
cd python-package
|
||||
|
||||
sed "s/CURRENT_DIR = os.path.dirname(__file__)/CURRENT_DIR = os.path.abspath(os.path.dirname(__file__))/g" -i setup.py
|
||||
sed "/^LIB_PATH.*/a LIB_PATH = [os.path.relpath(LIB_PATH[0], CURRENT_DIR)]" -i setup.py
|
||||
cat <<EOF >xgboost/libpath.py
|
||||
def find_lib_path():
|
||||
return ["${xgboost}/lib/${libname}"]
|
||||
EOF
|
||||
'';
|
||||
|
||||
postInstall = ''
|
||||
rm -rf $out/xgboost
|
||||
'';
|
||||
}
|
|
@ -20076,28 +20076,8 @@ EOF
|
|||
};
|
||||
};
|
||||
|
||||
xgboost = buildPythonPackage rec {
|
||||
name = "xgboost-${version}";
|
||||
|
||||
inherit (pkgs.xgboost) version src meta;
|
||||
|
||||
propagatedBuildInputs = with self; [ scipy ];
|
||||
checkInputs = with self; [ nose ];
|
||||
|
||||
postPatch = ''
|
||||
cd python-package
|
||||
|
||||
sed "s/CURRENT_DIR = os.path.dirname(__file__)/CURRENT_DIR = os.path.abspath(os.path.dirname(__file__))/g" -i setup.py
|
||||
sed "/^LIB_PATH.*/a LIB_PATH = [os.path.relpath(LIB_PATH[0], CURRENT_DIR)]" -i setup.py
|
||||
cat <<EOF >xgboost/libpath.py
|
||||
def find_lib_path():
|
||||
return ["${pkgs.xgboost}/lib/libxgboost.so"]
|
||||
EOF
|
||||
'';
|
||||
|
||||
postInstall = ''
|
||||
rm -rf $out/xgboost
|
||||
'';
|
||||
xgboost = callPackage ../development/python-modules/xgboost {
|
||||
xgboost = pkgs.xgboost;
|
||||
};
|
||||
|
||||
xkcdpass = buildPythonPackage rec {
|
||||
|
|
Loading…
Reference in a new issue