diff options
author | Daniel Bermond | 2018-11-22 15:16:31 +0000 |
---|---|---|
committer | Daniel Bermond | 2018-11-22 15:16:31 +0000 |
commit | fb148194b83bc65ea980b2b36863c0fe8d72eab7 (patch) | |
tree | 95c4260bb51bbca2e1067154aa2bd6774aaec208 /PKGBUILD | |
parent | 371e15aa8a0759b700c747e8b263451b5713c919 (diff) | |
download | aur-fb148194b83bc65ea980b2b36863c0fe8d72eab7.tar.gz |
Fix build (disable fbgemm). Add missing lib.
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 25 |
1 files changed, 13 insertions, 12 deletions
@@ -2,7 +2,7 @@ pkgname=caffe2-git _srcname=pytorch -pkgver=0.8.2.r14563.g4f622c26b9 +pkgver=0.8.2.r14826.gf79fb58744 pkgrel=1 epoch=1 pkgdesc='A new lightweight, modular, and scalable deep learning framework (git version)' @@ -182,7 +182,7 @@ build() { -DUSE_CUDA:BOOL='OFF' \ -DUSE_CUDNN:BOOL='OFF' \ -DUSE_DISTRIBUTED:BOOL='ON' \ - -DUSE_FBGEMM:BOOL='ON' \ + -DUSE_FBGEMM:BOOL='OFF' \ -DUSE_FFMPEG:BOOL='ON' \ -DUSE_GFLAGS:BOOL='ON' \ -DUSE_GLOG:BOOL='ON' \ @@ -200,6 +200,7 @@ build() { -DUSE_NNAPI:BOOL='OFF' \ -DUSE_NNPACK:BOOL='ON' \ -DUSE_NUMA:BOOL='ON' \ + -DUSE_NUMPY:BOOL='ON' \ -DUSE_NVRTC:BOOL='OFF' \ -DUSE_OBSERVERS:BOOL='ON' \ -DUSE_OPENCL:BOOL='OFF' \ @@ -231,17 +232,17 @@ package() { # remove unneeded files local _entry local _exclude_dirs - local _exclude_libs mapfile -t -d '' _exclude_dirs < <(find "${pkgdir}/usr/include" -mindepth 1 -maxdepth 1 -type d ! -name 'caffe*' -print0) - mapfile -t -d '' _exclude_libs < <(find -L "${pkgdir}/usr/lib" -maxdepth 1 -type f ! -name 'libcaffe*' -print0) - rm -f "$pkgdir"/usr/bin/{protoc,unzstd,zstd{cat,mt,}} - rm -f "$pkgdir"/usr/include/{*.h,*.py} - rm -rf "$pkgdir"/usr/lib/cmake/protobuf - rm -f "$pkgdir"/usr/lib/pkgconfig/{protobuf-lite,protobuf}.pc - rm -rf "$pkgdir"/usr/share/pkgconfig - rm -rf "$pkgdir"/usr/share/{ATen,cmake/{ATen,ONNX}} - rm -f "$pkgdir"/usr/share/man/man1/{unzstd,zstd{cat,}}.1 - for _entry in "${_exclude_dirs[@]}" "${_exclude_libs[@]}" + rm "$pkgdir"/usr/bin/{protoc,unzstd,zstd{cat,mt,}} + rm "$pkgdir"/usr/include/{*.h,*.py} + rm "$pkgdir"/usr/lib/*.a + rm "$pkgdir"/usr/lib/lib{zstd,onnxifi}* + rm -r "$pkgdir"/usr/lib/cmake/protobuf + rm "$pkgdir"/usr/lib/pkgconfig/{protobuf-lite,protobuf}.pc + rm "$pkgdir"/usr/share/pkgconfig/libzstd.pc + rm -r "$pkgdir"/usr/share/{ATen,cmake/{ATen,ONNX,Gloo}} + rm "$pkgdir"/usr/share/man/man1/{unzstd,zstd{cat,}}.1 + for _entry in "${_exclude_dirs[@]}" do rm -rf "$_entry" done |