summarylogtreecommitdiffstats
path: root/PKGBUILD
blob: d389e101613f167970a0a49af200ed8982a584ad (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
# Maintainer: Filip GraliƄski <filipg@amu.edu.pl>

pkgname=fairseq-git
pkgver=0.11
pkgrel=2
pkgdesc="Sequence modeling toolkit"
arch=('i686' 'x86_64')
url="https://github.com/pytorch/fairseq"
license=('MIT')
provides=('fairseq')
conflicts=('fairseq')
depends=('cuda'
         'nccl'
         'python-antlr4'
         'python-bitarray'
         'python-hydra'
         'python-omegaconf'
         'python-pytorch')
optdepends=('python-apex: Faster training')
source=("git+$url")
sha256sums=('SKIP')

build() {
    cd fairseq
    python setup.py build
}

package() {
    cd fairseq
    python setup.py install --root="$pkgdir" --optimize=1
}