blob: 415cc41b35d1957e559692ad46d5b959a8f4bc11 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
# Maintainer: redponike <proton (dot) me>
_pyname=xgboost
pkgname=python-xgboost-cuda
pkgver=2.1.4
pkgrel=1
pkgdesc="Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library. Compiled with CUDA support."
arch=('x86_64')
url="https://xgboost.ai"
license=('Apache-2.0')
provides=('python-xgboost')
conflicts=('python-xgboost' 'python-xgboost-git')
depends=(
'python-numpy'
'python-scipy'
'python-hypothesis'
'cuda'
'nccl'
)
makedepends=(
'python-build'
'python-installer'
'python-wheel'
'python-hatchling'
'python-setuptools'
'python-packaging'
'cmake'
'ninja'
'git'
)
checkdepends=('python-pytest')
optdepends=(
'apache-spark'
'python-pyspark: Distributed training support with PySpark'
'python-scikit-learn'
'python-pandas'
'python-matplotlib: Plotting with matplotlib'
'python-graphviz: Plotting with graphviz'
'python-dask: Dask support - Needs python-pandas and python-distributed'
'python-distributed: Dask support dependency'
'python-cloudpickle: PySpark support dependency'
'python-pyarrow'
)
source=(
$_pyname-$pkgver.tar.gz::https://github.com/dmlc/xgboost/archive/refs/tags/v$pkgver.tar.gz
git+https://github.com/dmlc/dmlc-core.git#commit=13341857549852a9a86b1894b5ba84c6276ab381
git+https://github.com/rapidsai/gputreeshap.git#commit=40eae8c4c45974705f8053e4d3d05b88e3cfaefd
)
sha256sums=('8c86edcdc4ea4e8ef92d62d71ddb59a2500d87312c5e1092cb41f60b59212101'
'8608303201f07bf3f7f8809e6db5bd64fbfcf964c0aceabbc4cf891271011cf7'
'0850233f9f4433bc2af8330a5c464a2cf8c43d2e46eaab875753157a100d8039')
prepare() {
cd "$srcdir/$_pyname-$pkgver"
rm -rf dmlc-core
ln -sf "$srcdir/dmlc-core" \
"$srcdir/$_pyname-$pkgver/dmlc-core"
rm -rf gputreeshap
ln -sf "$srcdir/gputreeshap" \
"$srcdir/$_pyname-$pkgver/gputreeshap"
}
build() {
cd "$srcdir"
cmake -B build -S $_pyname-$pkgver \
-D CMAKE_INSTALL_PREFIX=/usr \
-D BUILD_DEPRECATED_CLI=ON \
-D USE_CXX14_IF_AVAILABLE=ON \
-D FORCE_COLORED_OUTPUT=ON \
-D USE_OPENMP=ON \
-D USE_NCCL=ON \
-D USE_CUDA=ON \
-G Ninja \
-W no-dev
cmake --build build
cd $_pyname-$pkgver/python-package
python \
-m build \
--wheel \
--no-isolation \
--skip-dependency-check
}
check() {
cd $_pyname-$pkgver
python -m venv --system-site-packages test-env
source test-env/bin/activate
python -m installer python-package/dist/*.whl
python -m pytest -v tests/python -k 'not test_rabit_ops_ipv6'
deactivate
}
package() {
cd "$srcdir/$_pyname-$pkgver/python-package"
python \
-m installer dist/*.whl \
--destdir="$pkgdir" \
--compile-bytecode=2
}
|