diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 6 |
1 files changed, 2 insertions, 4 deletions
@@ -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 } |