initialize python-pytorch_scatter package

3 files changed, 101 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
new file mode 100644
index 000000000000..1f61bfa02ae9
--- /dev/null
+++ b/.SRCINFO
@@ -0,0 +1,36 @@
+pkgbase = python-pytorch_scatter
+	pkgdesc = PyTorch Extension Library of Optimized Scatter Operations
+	pkgver = 2.0.5
+	pkgrel = 1
+	url = https://github.com/rusty1s/pytorch_scatter
+	arch = any
+	license = MIT
+	depends = python
+	depends = cuda
+	depends = gcc
+	depends = cmake
+	depends = python-setuptools
+	depends = python-pytest
+	depends = python-pytest-cov
+	depends = python-pytorch
+	source = https://github.com/rusty1s/pytorch_scatter/archive/2.0.5.tar.gz
+	sha256sums = e29b364beaa9c84a99e0e236be89ed19d4452d89010ff736184ddcce488b47f6
+
+pkgname = python-pytorch_scatter
+
+pkgname = python-pytorch_scatter-cuda
+	pkgdesc = PyTorch Extension Library of Optimized Scatter Operations (with CUDA)
+	depends = python
+	depends = cuda
+	depends = gcc
+	depends = cmake
+	depends = python-setuptools
+	depends = python-pytest
+	depends = python-pytest-cov
+	depends = python-pytorch
+	depends = cuda
+	depends = cudnn
+	depends = python-pytorch-cuda
+	provides = python-pytorch_scatter
+	conflicts = python-pytorch_scatter
+
diff --git a/.gitignore b/.gitignore
new file mode 100644
index 000000000000..a600a28a6a80
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,3 @@
+*/
+*.tar.?z
+
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