summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: 48b639cf7819a06c5b94b6f261c0d40d68609fba (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
pkgname=python-torchdiffeq
pkgver=0.2.3
pkgrel=1
pkgdesc="Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation. "
depends=('python' 'python-pytorch' 'python-scipy')
makedepends=('python-setuptools')
arch=('x86_64')
url="https://github.com/rtqichen/torchdiffeq"
license=('MIT')
source=("https://files.pythonhosted.org/packages/61/7e/629146662e96da319fe237920b7d928a9832cbfa06c1d7ee8cdfe53ed450/torchdiffeq-0.2.3.tar.gz")
sha256sums=('fe75f434b9090ac0c27702e02bed21472b0f87035be6581f51edc5d4013ea31a')

build() {
  cd "${srcdir}/torchdiffeq-$pkgver"
  python setup.py build
}

package() {
  cd "${srcdir}/torchdiffeq-$pkgver"
  python setup.py install --root="$pkgdir" --optimize=1 --skip-build
}