PKGBUILD


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

# Maintainer: Scott Gigante <scottgigante@gmail.com>

_name=torch_spline_conv
_fullname=py$_name
pkgbase=python-$_fullname
pkgname=("python-$_fullname" "python-$_fullname-cuda")
pkgver=1.2.0
pkgrel=1
pkgdesc='Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch'
arch=(any)
url="https://github.com/rusty1s/$_fullname"
license=(MIT)
depends=(python gcc cuda cmake python-setuptools python-pytest python-pytest-cov python-pytorch)
_tarball="${_name}-${pkgver}.tar.gz"
source=("${url}/archive/${pkgver}.tar.gz")
sha256sums=('ab8da41357c8a4785662366655bb6dc5e84fd0e938008194955409aefe535009')
_srcdir="${_fullname}-${pkgver}"

prepare() {
    export CXX=g++
}

build() {
    cd $_srcdir
    python setup.py build
    cd ..
}

check() {
    cd $_srcdir
    python setup.py test
    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_spline_conv() {
  _package
}


package_python-pytorch_spline_conv-cuda() {
  pkgdesc="Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch (with CUDA)"
  depends+=(cuda cudnn python-pytorch-cuda)
  conflicts=(python-pytorch_spline_conv)
  provides=(python-pytorch_spline_conv)
  
  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
}