summarylogtreecommitdiffstats
path: root/PKGBUILD
diff options
context:
space:
mode:
Diffstat (limited to 'PKGBUILD')
-rw-r--r--PKGBUILD6
1 files changed, 2 insertions, 4 deletions
diff --git a/PKGBUILD b/PKGBUILD
index 3efae873d67e..0ad0751c4dcd 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -4,7 +4,7 @@ _CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6"
_pkgname=mxnet
pkgname=('mxnet-git' 'mxnet-cuda-git')
_pkgver=2.0.0
-pkgver=2.0.0.r11592.970a2cfbe7
+pkgver=2.0.0.r11611.dc69b04070
pkgrel=1
pkgdesc='A flexible and efficient library for deep learning'
arch=('x86_64')
@@ -96,9 +96,6 @@ build() {
cd "${srcdir}/mxnet-cuda-git/build"
cmake \
${cmake_opts[@]} \
- -DCMAKE_C_COMPILER=/opt/cuda/bin/gcc \
- -DCMAKE_CXX_COMPILER=/opt/cuda/bin/g++ \
- -DCMAKE_CUDA_HOST_COMPILER=/opt/cuda/bin/g++ \
-DMXNET_CUDA_ARCH=${_CUDA_ARCH_LIST} \
-DUSE_CUDA:BOOL=ON \
-DUSE_CUDNN:BOOL=ON \
@@ -150,6 +147,7 @@ _package() {
rm -rfv "${pkgdir}/usr/lib/cmake/dnnl"
rm -rfv "${pkgdir}/usr/lib/cmake/mkldnn"
rm -rfv "${pkgdir}/usr/share/doc"
+ rm -rfv "${pkgdir}/onednn"
find "${pkgdir}" -name '*.a' -delete
}