blob: f56ba1e174b1cca89f1119ed303fca77b48f73ae (
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
# Maintainer: Scott Cheng <aur@chengscott.io>
# Contributor: Alexander F. Rødseth <xyproto@archlinux.org>
# Contributor: flovo <flovo-git-commit@f-voegele.de>
pkgname=katago-cuda
_pkgname=katago
pkgver=1.14.1
pkgrel=2
pkgdesc='Baduk engine with no human-provided knowledge'
arch=(x86_64)
provides=($_pkgname)
conflicts=($_pkgname)
url='https://github.com/lightvector/KataGo'
license=(MIT)
depends=(libzip)
makedepends=(cmake cuda cudnn ninja)
source=("${pkgname}-${pkgver}.tar.gz"::"${url}/archive/refs/tags/v${pkgver}.tar.gz"
# Strongest confidently-rated network
"https://media.katagotraining.org/uploaded/networks/models/kata1/kata1-b18c384nbt-s9131461376-d4087399203.bin.gz"
# Latest network
"https://media.katagotraining.org/uploaded/networks/models/kata1/kata1-b18c384nbt-s9555257600-d4198296303.bin.gz")
sha256sums=('1a80d7fbd2b3a2684049afe61407d2276f6faf1dd1ca3f886cdb07c170c08e65'
'd266f19b4497ed19a72285a422813d8ee97e91d8a0b56f27585d0c02e61c9d38'
'30ff9a4f20f3f3cc40ba174187d87541e8becdd0dff54b12db7dfa24a3837d1e')
_model_source=("${source[@]:1}")
_model=("${_model_source[@]##*/}")
noextract=("${_model[@]}")
build() {
cmake \
-S "KataGo-${pkgver}/cpp" \
-B build \
-D CMAKE_BUILD_TYPE=Release \
-D CMAKE_INSTALL_PREFIX=/usr \
-D USE_BACKEND=CUDA \
-D NO_GIT_REVISION=1 \
-G Ninja
cmake --build build
}
package() {
install -Dm755 build/katago "${pkgdir}/usr/bin/katago"
install -Dm644 "KataGo-${pkgver}/LICENSE" "${pkgdir}/usr/share/licenses/${_pkgname}/LICENSE"
# install models
for mi in "${_model[@]}"; do
install -Dm644 "${mi}" "${pkgdir}/usr/share/${_pkgname}/${mi}"
done
}
|