blob: 1c08965b27a0e7557fc47455f058a875de94c8b1 (
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
|
# Maintainer: Leo Mao <leomaoyw at gmail dot com>
pkgname=python-cupy
_pkgname=cupy
pkgver=10.6.0
_cub_commit=c3cceac115c072fb63df1836ff46d8c60d9eb304
_jitify_commit=4a37de0be4639f222c6565ebd0654cb922b5180e
pkgrel=1
pkgdesc="NumPy-like API accelerated with CUDA"
url="https://cupy.dev"
arch=('x86_64')
license=('MIT')
depends=('cuda' 'python-fastrlock' 'python-numpy')
makedepends=('cudnn' 'cython' 'nccl' 'python-setuptools')
optdepends=('cudnn' 'libcutensor' 'nccl')
source=("https://github.com/cupy/cupy/archive/v$pkgver.tar.gz"
"https://github.com/NVIDIA/cub/archive/$_cub_commit.tar.gz"
"https://github.com/NVIDIA/jitify/archive/$_jitify_commit.tar.gz")
md5sums=('63693ad1ca502a8919d72d089e8b216f'
'ae6435aef98378a8b323b69f6665df33'
'2ad752c0814c2da9909e2dcac0f50401')
prepare() {
cd "$srcdir/$_pkgname-$pkgver"
rmdir cupy/_core/include/cupy/cub
ln -srfT "$srcdir/cub-$_cub_commit" cupy/_core/include/cupy/cub
rmdir cupy/_core/include/cupy/jitify
ln -srfT "$srcdir/jitify-$_jitify_commit" cupy/_core/include/cupy/jitify
}
build() {
cd "$srcdir/$_pkgname-$pkgver"
CUDA_PATH=/opt/cuda python setup.py build
}
package() {
cd "$srcdir/$_pkgname-$pkgver"
python setup.py install --skip-build --prefix=/usr --root="$pkgdir" --optimize=1
install -Dm644 LICENSE "${pkgdir}/usr/share/licenses/${pkgname}/LICENSE"
}
|