blob: d06cce3da9f668d5d34d3188135bda8a02ad657a (
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
|
# Maintainer: crl <crl18039102576@126.com>
pkgbase=rmm
pkgname=(rmm python-rmm)
pkgver=26.08.00
pkgrel=1
pkgdesc="RAPIDS Memory Manager"
url="https://github.com/rapidsai/rmm"
arch=('x86_64')
license=('Apache-2.0')
depends=('libgcc' 'libstdc++' 'cuda' 'nvtx' 'rapids-logger')
makedepends=('cmake' 'gcc' 'ninja' 'cython' 'python-build' 'python-installer' 'python-wheel' 'python-rapids-build-backend')
source=(
"$url/archive/refs/tags/v$pkgver.tar.gz"
"system-lib.patch"
"missing-pkg.patch"
)
sha256sums=('6c568f0999323977c1c808e2cc9d4ab20153b5fca6bab1072801b45638ec40c9'
'61ea872a61846518804b698842d4d9b01f09847cc9b58480337173b5b6d8a509'
'1c5d8c1f36bea1934d778124c8fb79a27e4f1b8e408e18a8afc7d5309acdac58')
prepare() {
cd "$srcdir/$pkgname-$pkgver"
patch -p1 cpp/CMakeLists.txt < "$srcdir/system-lib.patch"
patch -p1 python/rmm/CMakeLists.txt < "$srcdir/missing-pkg.patch"
}
build() {
cd "$srcdir/$pkgname-$pkgver"
export CXXFLAGS="$CXXFLAGS -DCCCL_IGNORE_DEPRECATED_STREAM_REF_HEADER"
cmake -B build -S cpp \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_INSTALL_PREFIX=/usr \
-DBUILD_TESTS=OFF \
-DRMM_NVTX=ON \
-G Ninja
cmake --build build
cd python/rmm
export rmm_DIR="$srcdir/$pkgname-$pkgver/build"
python -m build --wheel --no-isolation --skip-dependency-check
}
package_rmm() {
cd "$srcdir/$pkgname-$pkgver"
DESTDIR="$pkgdir" cmake --install build
}
package_python-rmm() {
depends+=('python-cuda' 'rmm')
cd "$srcdir/$pkgbase-$pkgver/python/rmm"
python -m installer --destdir="$pkgdir" dist/*.whl
}
|