blob: 61d12666ea3f4d0d4cca417571d65dd33c83b884 (
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
|
# Maintainer: Jakub OkoĊski <jakub@okonski.org>
pkgname=miopen
pkgver=1.7.1
pkgrel=1
pkgdesc="AMD's Machine Intelligence Library"
arch=('x86_64')
url="https://github.com/ROCmSoftwarePlatform/MIOpen"
license=('NCSAOSL')
depends=(ocl-icd hip hcc rocblas "boost>=1.58" clang-ocl)
makedepends=(git opencl-headers cmake gcc ninja ocl-icd hip hcc rocblas "boost>=1.58" miopengemm)
source=("git+https://github.com/ROCmSoftwarePlatform/MIOpen.git#tag=1.7.1")
md5sums=('SKIP')
build() {
mkdir -p "$srcdir/build"
cd "$srcdir/build"
# compile with HCC
export CXX=/opt/rocm/hcc/bin/hcc
# TODO: fix libMIOpen.so, it contains references to $srcdir
cmake -DCMAKE_BUILD_TYPE=Release \
-DCMAKE_INSTALL_PREFIX="$pkgdir/opt/rocm/miopen" \
-DMIOPEN_BACKEND=HIP \
-DCMAKE_PREFIX_PATH="/opt/rocm/hcc;/opt/rocm/hip" \
-DHALF_INCLUDE_DIR="$startdir" \
-G Ninja \
"$srcdir/MIOpen"
ninja
}
package() {
ninja -C "$srcdir/build" install
mkdir -p $pkgdir/etc/ld.so.conf.d
cat <<-EOF > $pkgdir/etc/ld.so.conf.d/miopen.conf
/opt/rocm/miopen/lib/
EOF
}
|