diff options
author | Scott Gigante | 2021-01-07 18:50:17 -0500 |
---|---|---|
committer | Scott Gigante | 2021-01-07 18:50:17 -0500 |
commit | 0c16f649ca12f05ebbb17421f0d24933e391bf96 (patch) | |
tree | 7cba1fb0ea0a43a034b7af7678c4f1938a615e0d /PKGBUILD | |
download | aur-python-pytorch_scatter.tar.gz |
initialize python-pytorch_scatter package
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/PKGBUILD b/PKGBUILD new file mode 100644 index 000000000000..98bf429b1dd6 --- /dev/null +++ b/PKGBUILD @@ -0,0 +1,62 @@ +# Maintainer: Scott Gigante <scottgigante@gmail.com> + +_name=torch_scatter +_fullname=py$_name +pkgbase=python-$_fullname +pkgname=("python-$_fullname" "python-$_fullname-cuda") +pkgver=2.0.5 +pkgrel=1 +pkgdesc='PyTorch Extension Library of Optimized Scatter Operations' +arch=(any) +url="https://github.com/rusty1s/$_fullname" +license=(MIT) +depends=(python cuda gcc cmake python-setuptools python-pytest python-pytest-cov python-pytorch) +_tarball="${_name}-${pkgver}.tar.gz" +source=("${url}/archive/${pkgver}.tar.gz") +sha256sums=('e29b364beaa9c84a99e0e236be89ed19d4452d89010ff736184ddcce488b47f6') +_srcdir="${_fullname}-${pkgver}" + +prepare() { + export CXX=g++ +} + +build() { + cd $_srcdir + python setup.py build + cd .. +} + +check() { + cd $_srcdir + python setup.py test --addopts "-k 'not segment'" # skip failing tests + cd .. +} + +_package() { + cd $_srcdir + python setup.py bdist_wheel + local site="$pkgdir/usr/lib/$(readlink /bin/python3)/site-packages" + mkdir -p "$site" + unzip "$(ls -1 dist/${_name}-${pkgver}-*.whl)" -d "$site" + rm -r ${site}/test + cd .. +} + + +package_python-pytorch_scatter() { + _package +} + + +package_python-pytorch_scatter-cuda() { + pkgdesc="PyTorch Extension Library of Optimized Scatter Operations (with CUDA)" + depends+=(cuda cudnn python-pytorch-cuda) + conflicts=(python-pytorch_scatter) + provides=(python-pytorch_scatter) + + export CUDA_HOME=/opt/cuda + export PATH=$CUDA_HOME/bin:$PATH + export CPATH=$CUDA_HOME/include:$CPATH + export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH + _package +}
\ No newline at end of file |