blob: 4990690506e06859b25a638e0c805510cbf3ba6b (
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
# Maintainer: Gustavo Alvarez <sl1pkn07@gmail.com>
pkgbase=libcutensor
pkgname=('libcutensor'
'libcutensor-common'
'libcutensor-cuda10.1'
'libcutensor-cuda10.2'
)
pkgver=1.2.2.5
_pkg_file_name="libcutensor-linux-x86_64-${pkgver}.tar.gz"
pkgrel=1
pkgdesc="GPU-accelerated tensor linear algebra library"
arch=('x86_64')
url='https://developer.nvidia.com/cutensor'
license=('custom:NVIDIA')
_DOWNLOADS_DIR=`xdg-user-dir DOWNLOAD`
if [ ! -f ${PWD}/${_pkg_file_name} ]; then
if [ -f $_DOWNLOADS_DIR/${_pkg_file_name} ]; then
ln -sfn $_DOWNLOADS_DIR/${_pkg_file_name} ${PWD}
else
msg2 ""
msg2 "The package can be downloaded here: "
msg2 "Please remember to put a downloaded package ${_pkg_file_name} into the build directory ${PWD} or $_DOWNLOADS_DIR"
msg2 ""
fi
fi
source=("local://${_pkg_file_name}")
sha256sums=('SKIP')
_create_links() {
# create soname links
for _lib in $(find "${pkgdir}" -name '*.so*'); do
_soname=$(dirname "${_lib}")/$(readelf -d "${_lib}" | grep -Po 'SONAME.*: \[\K[^]]*' || true)
_base=$(echo ${_soname} | sed -r 's/(.*).so.*/\1.so/')
[[ -e "${_soname}" ]] || ln -s $(basename "${_lib}") "${_soname}"
[[ -e "${_base}" ]] || ln -s $(basename "${_soname}") "${_base}"
done
}
package_libcutensor() {
depends=('gcc-libs'
'cuda'
'libcutensor-common'
)
cd libcutensor
install -Dm644 "lib/11/libcutensor_static.a" "${pkgdir}/opt/cuda/x86_64-linux/lib/libcutensor_static.a"
install -Dm755 "lib/11/libcutensor.so.${pkgver//.5/}" "${pkgdir}/opt/cuda/targets/x86_64-linux/lib/libcutensor.so.${pkgver//.5/}"
_create_links
install -d "${pkgdir}/usr/share/licenses/${pkgname}"
ln -s "${pkgdir}/usr/share/licenses/libcutensor-common/license.pdf" "${pkgdir}/usr/share/licenses/${pkgname}/license.pdf"
}
package_libcutensor-cuda10.1() {
pkgdesc="GPU-accelerated tensor linear algebra library (CUDA 10.1)"
depends=('gcc-libs'
'cuda-10.1'
'libcutensor-common'
)
cd libcutensor
install -Dm644 "lib/10.1/libcutensor_static.a" "${pkgdir}/opt/cuda-10.1/x86_64-linux/lib/libcutensor_static.a"
install -Dm755 "lib/10.1/libcutensor.so.${pkgver%%.[0-9]}" "${pkgdir}/opt/x86_64-linux/lib/libcutensor.so.${pkgver%%.[0-9]}"
_create_links
install -d "${pkgdir}/usr/share/licenses/${pkgname}"
ln -s "${pkgdir}/usr/share/licenses/libcutensor-common/license.pdf" "${pkgdir}/usr/share/licenses/${pkgname}/license.pdf"
}
package_libcutensor-cuda10.2() {
pkgdesc="GPU-accelerated tensor linear algebra library (CUDA 10.2)"
depends=('gcc-libs'
'cuda-10.2'
'libcutensor-common'
)
cd libcutensor
install -Dm644 "lib/10.2/libcutensor_static.a" "${pkgdir}/opt/cuda-10.2/x86_64-linux/lib/libcutensor_static.a"
install -Dm755 "lib/10.2/libcutensor.so.${pkgver%%.[0-9]}" "${pkgdir}/opt/cuda-10.2/x86_64-linux/lib/libcutensor.so.${pkgver%%.[0-9]}"
_create_links
install -d "${pkgdir}/usr/share/licenses/${pkgname}"
ln -s "${pkgdir}/usr/share/licenses/libcutensor-common/license.pdf" "${pkgdir}/usr/share/licenses/${pkgname}/license.pdf"
}
package_libcutensor-common() {
pkgdesc="GPU-accelerated tensor linear algebra library (Common bits)"
arch=('any')
cd libcutensor
for i in $(find include -type f); do install -Dm644 ${i} "${pkgdir}/usr/${i}"; done
install -Dm644 license.pdf "${pkgdir}/usr/share/licenses/${pkgname}/license.pdf"
}
|